Iker commited on
Commit
a4d80df
1 Parent(s): ad48936

Batch size description

Browse files
Files changed (1) hide show
  1. translate.py +2 -1
translate.py CHANGED
@@ -190,7 +190,8 @@ if __name__ == "__main__":
190
  "--starting_batch_size",
191
  type=int,
192
  default=128,
193
- help="Starting batch size, we will automatically reduce it if we find an OOM error.",
 
194
  )
195
 
196
  parser.add_argument(
 
190
  "--starting_batch_size",
191
  type=int,
192
  default=128,
193
+ help="Starting batch size, we will automatically reduce it if we find an OOM error."
194
+ "If you use multiple devices, we will divide this number by the number of devices.",
195
  )
196
 
197
  parser.add_argument(