yml change

This commit is contained in:
sunny
2024-10-29 15:27:42 -04:00
parent 46d2b4ce89
commit fc1f275e6c

View File

@@ -254,7 +254,7 @@ unfrozen_parameters:
# model.embed_tokens layers
gradient_accumulation_steps: 16
gradient_accumulation_steps: 2
micro_batch_size: 2
num_epochs: 3
optimizer: adamw_torch_fused