1
1
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user