Open deneschen opened 1 year ago
发现你的代码里面写错了。。。和数据文件名字不匹配。
diff --git a/CodeReviewer/code/run_finetune_msg.py b/CodeReviewer/code/run_finetune_msg.py
index 865530b..d6cf954 100644
--- a/CodeReviewer/code/run_finetune_msg.py
+++ b/CodeReviewer/code/run_finetune_msg.py
@@ -169,7 +169,7 @@ def main(args):
train_file = args.train_filename
valid_file = args.dev_filename
if os.path.isdir(train_file):
- train_files = [file for file in os.listdir(train_file) if file.startswith("train") and file.endswith(".jsonl")]
+ train_files = [file for file in os.listdir(train_file) if file.startswith("msg-train") and file.endswith(".jsonl")]
else:
train_files = [train_file]
random.seed(args.seed)
是的,如果你发现了代码中的bug,请提交pull request帮助我们修改,谢谢!
为什么这个脚本finetune-msg.sh没有生成checkpoints的step,而是直接finished了是我哪里修改问题吗?
下面是执行的日志:
我的目录结构是这样子的