Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Reed Wanderman-Milne's avatar
    Do not expose --max_train_steps in models that do not use it. · 824ff2d6
    Reed Wanderman-Milne authored 
    Only the V1 resnet model uses --max_train_steps. This unexposes the flag in the keras_application_models, mnist, keras resnet, CTL resnet Models. Before this change, such models allowed the flag to be specified, but ignored it.

I also removed the "max_train" argument from the run_synthetic function, since this only had any meaning for the V1 resnet model. Instead, the V1 resnet model now directly passes --max_train_steps=1 to run_synthetic.

PiperOrigin-RevId: 264269836
    824ff2d6
_performance.py 12.1 KB