1

2024-12-02 00:20:51 +08:00
parent adeb153c8a
commit f1836172d6
4 changed files with 19 additions and 19 deletions
--- a/scripts/train_dizhu_model.py
+++ b/scripts/train_dizhu_model.py
@@ -23,7 +23,7 @@ def train_dizhu_model():

    # 训练模型，设定总训练步数
    logger.info("开始训练斗地主模型...")
-    model.learn(total_timesteps=100000)  # 总训练步数
+    model.learn(total_timesteps=10000000000000000)  # 总训练步数
    logger.info("斗地主模型训练完成！")

    # 保存训练后的模型