瀏覽代碼

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

zhouyang.xie 3 月之前
父節點
當前提交
8d19984abd
共有 1 個文件被更改,包括 1 次插入1 次删除
  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")
 
     # 设置环境变量
     """