Update args.json
Browse files
args.json
CHANGED
@@ -24,7 +24,7 @@
|
|
24 |
"train_batch_size": 1,
|
25 |
"sample_batch_size": 4,
|
26 |
"num_train_epochs": 16,
|
27 |
-
"max_train_steps":
|
28 |
"gradient_accumulation_steps": 1,
|
29 |
"gradient_checkpointing": false,
|
30 |
"learning_rate": 1e-06,
|
|
|
24 |
"train_batch_size": 1,
|
25 |
"sample_batch_size": 4,
|
26 |
"num_train_epochs": 16,
|
27 |
+
"max_train_steps": 500,
|
28 |
"gradient_accumulation_steps": 1,
|
29 |
"gradient_checkpointing": false,
|
30 |
"learning_rate": 1e-06,
|