mlcommons / training

Reference implementations of MLPerf™ training benchmarks
https://mlcommons.org/en/groups/training
Apache License 2.0
1.62k stars 560 forks source link

Command line options in bert training #688

Closed mahmoodn closed 3 months ago

mahmoodn commented 1 year ago

Hi In the readme file of language_model/bert, I see the final training command has the following options:

  --do_eval \
  --nodo_train \

I was wondering what does that mean exactly? So, is that really a training command?

ShriyaPalsamudram commented 3 months ago

Passing --nodo_eval and --do_train ensuring that the reference does only training. Switching it to --do_eval --nodo_train runs in eval-only mode.

Running the model section in the README shares commands for both training and evaluation of the model using the right args.