AlexeyGHT commited on
Commit
dea9f51
·
verified ·
1 Parent(s): 011b70c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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=40,
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,