From 416cc90b58fbbe37c49e82899f156c07eddbd5f1 Mon Sep 17 00:00:00 2001
From: gongjy <2474590974@qq.com>
Date: Wed, 12 Feb 2025 20:34:47 +0800
Subject: [PATCH] update ckp-path

---
 train_dpo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_dpo.py b/train_dpo.py
index 2c857ff..dc9054a 100644
--- a/train_dpo.py
+++ b/train_dpo.py
@@ -134,7 +134,7 @@ def init_model(lm_config):
     tokenizer = AutoTokenizer.from_pretrained('./model/minimind_tokenizer')
     model = MiniMindLM(lm_config)
     moe_path = '_moe' if lm_config.use_moe else ''
-    ckp = f'./out/full_dist_{lm_config.dim}{moe_path}.pth'
+    ckp = f'./out/full_sft_{lm_config.dim}{moe_path}.pth'
     state_dict = torch.load(ckp, map_location=args.device)
     model.load_state_dict(state_dict, strict=False)
     # 初始化参考模型