Get qlora mistral-7b fine tuning working on a single 4090 (#708)
This commit is contained in:
@@ -43,7 +43,7 @@ wandb_run_id:
|
||||
wandb_log_model:
|
||||
|
||||
gradient_accumulation_steps: 4
|
||||
micro_batch_size: 4
|
||||
micro_batch_size: 2
|
||||
num_epochs: 1
|
||||
optimizer: adamw_bnb_8bit
|
||||
lr_scheduler: cosine
|
||||
|
||||
Reference in New Issue
Block a user