liamcloud commited on
Commit
708e5b5
1 Parent(s): 736daaa

Update args.json

Browse files
Files changed (1) hide show
  1. args.json +1 -1
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": 800,
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,