@@ -192,12 +192,10 @@ python run_squad.py --help
192192Aside from options to set hyperparameters, the relevant options to control the behaviour of the ` run_pretraining.py ` script are:
193193``` bash
194194 --[no]amp: Whether to enable AMP ops.(default: ' false' )
195- --[no]amp_fastmath: Whether to enable AMP fasthmath ops.(default: ' false' )
196195 --bert_config_file: The config json file corresponding to the pre-trained BERT model. This specifies the model architecture.
197196 --[no]do_eval: Whether to run evaluation on the dev set.(default: ' false' )
198197 --[no]do_train: Whether to run training.(evaluation: ' false' )
199198 --eval_batch_size: Total batch size for eval.(default: ' 8' )(an integer)
200- --[no]fastmath: Whether to enable loss scaler for fasthmath ops.(default: ' false' )
201199 --[no]horovod: Whether to use Horovod for multi-gpu runs(default: ' false' )
202200 --init_checkpoint: Initial checkpoint (usually from a pre-trained BERT model).
203201 --input_file: Input TF example files (can be a glob or comma separated).
@@ -255,24 +253,12 @@ Where:
255253
256254- <eval_batch_size> per-gpu batch size used for evaluation after training.<learning_rate> Default rate of 1e-4 is good for global batch size 256.
257255
258- - <precision > Type of math in your model, can be either fp32, fp16, fp16_xla, fastmath, amp_fm, amp_fm_xla, amp or amp_xla . The options mean:
256+ - <precision > Type of math in your model, can be either fp32, or amp . The options mean:
259257
260258 - fp32 32 bit IEEE single precision floats.
261259
262- - fp16 Hand-coded mixed precision 16 and 32 bit floats.
263-
264- - fp16 Hand-coded mixed precision floats, JIT compiled with XLA.
265-
266- - fastmath Matmuls done by tensor cores in mixed precision, the rest is done in FP32.
267-
268- - amp_fm Alternative FastMath implementation that works by manipulating TensorFlow’s compute graph.
269-
270- - amp_fm_xla The amp_fm flag plus XLA JIT compilation.
271-
272260 - amp Automatic rewrite of TensorFlow compute graph to take advantage of 16 bit arithmetic whenever that is safe.
273261
274- - amp_xla The amp flag plus XLA JIT compilation.
275-
276262- <num_gpus> Number of GPUs to use for training. Must be equal to or smaller than the number of GPUs attached to your node.
277263
278264- <warmup_steps> Number of warm-up steps at the start of training.
0 commit comments