PathFinderKR commited on
Commit
9287b58
1 Parent(s): 64aa228

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -179,10 +179,9 @@ bias="none"
179
  ################################################################################
180
  # TrainingArguments parameters
181
  ################################################################################
182
- num_train_epochs=1
183
  per_device_train_batch_size=1
184
- per_device_eval_batch_size=2
185
- gradient_accumulation_steps=4
186
  gradient_checkpointing=True
187
  learning_rate=2e-5
188
  lr_scheduler_type="cosine"
 
179
  ################################################################################
180
  # TrainingArguments parameters
181
  ################################################################################
182
+ num_train_epochs=2
183
  per_device_train_batch_size=1
184
+ gradient_accumulation_steps=1
 
185
  gradient_checkpointing=True
186
  learning_rate=2e-5
187
  lr_scheduler_type="cosine"