Skip to content

GitLab

"lm_eval/tasks/squadv2/task.py" did not exist on "8ac992696564547de95655ec250d16dc4bd41782"
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
census_test.py 5.86 KB