From b57ebc76c66edfde8c489f6125122b10e5a2f685 Mon Sep 17 00:00:00 2001 From: Dixi Yao Date: Wed, 8 Nov 2023 21:06:55 -0500 Subject: [PATCH] used the checkpoint path as the temporary path for huggingface trainer. --- .../llm_split_learning/split_learning_trainer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/examples/split_learning/llm_split_learning/split_learning_trainer.py b/examples/split_learning/llm_split_learning/split_learning_trainer.py index f73a8ac44..7082394d1 100644 --- a/examples/split_learning/llm_split_learning/split_learning_trainer.py +++ b/examples/split_learning/llm_split_learning/split_learning_trainer.py @@ -104,7 +104,10 @@ def __init__(self, model=None, callbacks=None): parser = HfArgumentParser(TrainingArguments) (self.training_args,) = parser.parse_args_into_dataclasses( - args=["--report_to=none"] + args=[ + "--output_dir=" + Config.params["checkpoint_path"], + "--report_to=none", + ] ) # Redesign the evaluation stage.