Update README.md
Browse files
README.md
CHANGED
|
@@ -11,7 +11,7 @@ args = Namespace(
|
|
| 11 |
lr=1e-5,
|
| 12 |
gradient_accumulation_steps=1,
|
| 13 |
train_batch_size=64,
|
| 14 |
-
num_train_epochs=
|
| 15 |
lr_scheduler="constant",
|
| 16 |
lr_warmup_steps=1,
|
| 17 |
snr_gamma=5,
|
|
|
|
| 11 |
lr=1e-5,
|
| 12 |
gradient_accumulation_steps=1,
|
| 13 |
train_batch_size=64,
|
| 14 |
+
num_train_epochs=20,
|
| 15 |
lr_scheduler="constant",
|
| 16 |
lr_warmup_steps=1,
|
| 17 |
snr_gamma=5,
|