merge and sync with main before cherry-pick

sotopia-lab · Mar 14, 2024 · 57db89c · 57db89c
1 parent ce15cae
commit 57db89c
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 211 deletions.
diff --git a/llm_self_train/check_gcp_monitor_download.py b/llm_self_train/check_gcp_monitor_download.py
diff --git a/llm_self_train/gcp_demo.py b/llm_self_train/gcp_demo.py
diff --git a/llm_self_train/main.py b/llm_self_train/main.py
diff --git a/llm_self_train/resources/train_args.yml b/llm_self_train/resources/train_args.yml
@@ -29,7 +29,8 @@ overwrite_cache: true
 output_dir: NEED-CONFIG
 overwrite_output_dir: true
 logging_steps: 1
-save_strategy: "epoch"
+save_strategy: "steps"
+save_steps: 1000000000
 save_total_limit: 5
 use_auth_token: True
 wandb_token: NEED-CONFIG