Forráskód Böngészése

遵循面向对象思想重构train_model_grpo.py

zhouyang.xie 3 hónapja
szülő
commit
8d19984abd
1 módosított fájl, 1 hozzáadás és 1 törlés
  1. 1 1
      src/train_model_grpo_v1.py

+ 1 - 1
src/train_model_grpo_v1.py

@@ -215,7 +215,7 @@ class ModelTrainer:
 
 if __name__ == "__main__":
     # 加载配置文件
-    config = load_config(None)
+    config = load_config(f"../conf/conf_train.yaml")
 
     # 设置环境变量
     """