mirror/OpenLLM
1
0
Fork 0
mirror of https://github.com/bentoml/OpenLLM.git synced 2026-04-25 09:32:37 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(ft): set report_to none to avoid wandb setup

Browse Source 
Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
aarnphm-ec2-dev
2023-07-21 18:33:38 +00:00
parent 8a42832360
commit beb8c2bb08
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/openllm/playground/llama2_qlora.py
View File

@@ -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")