1

2024-12-01 22:14:23 +08:00
parent 5eef2384cf
commit 3e65e02704
29 changed files with 87 additions and 39 deletions
--- a/scripts/train_chengdu_mahjong_model.py
+++ b/scripts/train_chengdu_mahjong_model.py
@@ -1,6 +1,5 @@
-import gym
 from stable_baselines3 import PPO
-from src.environment.chengdu_mahjong_env import ChengduMahjongEnv
+from src import ChengduMahjongEnv
 import torch
 from configs.log_config import setup_logging
 from loguru import logger  # 添加 logger
--- a/scripts/train_dizhu_model.py
+++ b/scripts/train_dizhu_model.py