PathFinderKR
commited on
Commit
•
9287b58
1
Parent(s):
64aa228
Update README.md
Browse files
README.md
CHANGED
@@ -179,10 +179,9 @@ bias="none"
|
|
179 |
################################################################################
|
180 |
# TrainingArguments parameters
|
181 |
################################################################################
|
182 |
-
num_train_epochs=
|
183 |
per_device_train_batch_size=1
|
184 |
-
|
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"
|