From beb8c2bb080093119b2f51ce3fdb296bd8d4f02c Mon Sep 17 00:00:00 2001 From: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com> Date: Fri, 21 Jul 2023 18:33:38 +0000 Subject: [PATCH] fix(ft): set report_to none to avoid wandb setup Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com> --- src/openllm/playground/llama2_qlora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/openllm/playground/llama2_qlora.py b/src/openllm/playground/llama2_qlora.py index 5351a451..e0367c96 100644 --- a/src/openllm/playground/llama2_qlora.py +++ b/src/openllm/playground/llama2_qlora.py @@ -162,7 +162,7 @@ class TrainingArguments: learning_rate: float = dataclasses.field(default=5e-5) num_train_epochs: int = dataclasses.field(default=3) logging_steps: int = dataclasses.field(default=1) - logging_strategy: str = dataclasses.field(default="steps") + report_to: str = dataclasses.field(default="none") output_dir: str = dataclasses.field(default=os.path.join(os.getcwd(), "outputs", "llama")) save_strategy: str = dataclasses.field(default="no")