mtasic85 commited on
Commit
c162f45
1 Parent(s): 81565a0

pretrain model

Browse files
Files changed (1) hide show
  1. scripts/pretrain-model.yaml +1 -1
scripts/pretrain-model.yaml CHANGED
@@ -69,7 +69,7 @@ train:
69
  global_batch_size: 512
70
 
71
  # Number of samples per data-parallel rank (type: int, default: 4)
72
- micro_batch_size: 16
73
 
74
  # Number of iterations with learning rate warmup active (type: int, default: 2000)
75
  # lr_warmup_steps: 2000
 
69
  global_batch_size: 512
70
 
71
  # Number of samples per data-parallel rank (type: int, default: 4)
72
+ micro_batch_size: 17
73
 
74
  # Number of iterations with learning rate warmup active (type: int, default: 2000)
75
  # lr_warmup_steps: 2000