From 84fc217f7917d760ab2f4c4c4707d8644d7e7fb3 Mon Sep 17 00:00:00 2001 From: Viktorius Suwandi Date: Mon, 29 May 2023 15:41:24 +0700 Subject: [PATCH] Update wandb_log_model on gpt_neox_20b.yml --- configs/gpt_neox_20b.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/gpt_neox_20b.yml b/configs/gpt_neox_20b.yml index 91698ffaa..260b648b5 100644 --- a/configs/gpt_neox_20b.yml +++ b/configs/gpt_neox_20b.yml @@ -23,7 +23,7 @@ lora_fan_in_fan_out: true # pythia/GPTNeoX lora specific wandb_project: gpt4all-neox-20b wandb_watch: wandb_run_id: -wandb_log_model: checkpoint +wandb_log_model: output_dir: ./gpt4all-neox-20b batch_size: 48 micro_batch_size: 4