Skip to content

Commit ec9eaf7

Browse files
author
Swetha Mandava
committed
removing fast_math, etc from README
1 parent 42c6a14 commit ec9eaf7

1 file changed

Lines changed: 1 addition & 15 deletions

File tree

TensorFlow/LanguageModeling/BERT/README.md

Lines changed: 1 addition & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -192,12 +192,10 @@ python run_squad.py --help
192192
Aside 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

Comments
 (0)