1
This commit is contained in:
2024-11-30 21:04:03 +08:00
parent fd6006b186
commit 4142bd9423
6 changed files with 70 additions and 10 deletions

View File

@@ -2,6 +2,7 @@ import gym
from stable_baselines3 import PPO
from src.environment.chengdu_majiang_env import MahjongEnv
import torch
from configs.log_config import setup_logging
def train_model():
# 创建 MahjongEnv 环境实例
@@ -29,4 +30,6 @@ def train_model():
env.render() # 打印环境状态
if __name__ == "__main__":
# 调用配置函数来设置日志
setup_logging()
train_model()