From abddcf4dfee2db8c8117b2c43f5bec81ab518626 Mon Sep 17 00:00:00 2001 From: Viktorius Suwandi Date: Mon, 29 May 2023 16:31:53 +0700 Subject: [PATCH] Update wandb_log_model on pythia_1_2B_alpaca.yml --- configs/pythia_1_2B_alpaca.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/pythia_1_2B_alpaca.yml b/configs/pythia_1_2B_alpaca.yml index 5047be6bf..aa0587b90 100644 --- a/configs/pythia_1_2B_alpaca.yml +++ b/configs/pythia_1_2B_alpaca.yml @@ -26,7 +26,7 @@ lora_fan_in_fan_out: true # pythia/GPTNeoX lora specific wandb_project: pythia-1.4b-lora wandb_watch: wandb_run_id: -wandb_log_model: checkpoint +wandb_log_model: output_dir: ./lora-alpaca batch_size: 48 micro_batch_size: 4