finetune-msg.sh no step to generate checkpoints? about codebert

[deneschen]

发现你的代码里面写错了。。。和数据文件名字不匹配。

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)

[celbree]

是的，如果你发现了代码中的bug，请提交pull request帮助我们修改，谢谢！