diff --git a/configs/vicuna_13B_4bit_reflect.yml b/configs/vicuna_13B_4bit_reflect.yml index 872ce59ca..7ad409f26 100644 --- a/configs/vicuna_13B_4bit_reflect.yml +++ b/configs/vicuna_13B_4bit_reflect.yml @@ -28,7 +28,7 @@ lora_fan_in_fan_out: false wandb_project: wandb_watch: wandb_run_id: -wandb_log_model: checkpoint +wandb_log_model: output_dir: ./lora-reflect batch_size: 8 micro_batch_size: 2