GitLab

PiperOrigin-RevId: 264911876

PiperOrigin-RevId: 264895439

PiperOrigin-RevId: 264867908

PiperOrigin-RevId: 264853703

PiperOrigin-RevId: 264776396

PiperOrigin-RevId: 264647492

PiperOrigin-RevId: 264527204

PiperOrigin-RevId: 264511778

PiperOrigin-RevId: 264500330

PiperOrigin-RevId: 264474346

PiperOrigin-RevId: 264470356

PiperOrigin-RevId: 264468498

PiperOrigin-RevId: 264411162

PiperOrigin-RevId: 264394112

PiperOrigin-RevId: 264300408

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

Moving Keras ResNet models to `official/vision/image_classification` and benchmarks to `official/benchmark`.

PiperOrigin-RevId: 264268533

An earlier change made it so running this file would cause an error. The unit tests still passed, as the unit tests do not directly call the main() function

PiperOrigin-RevId: 264268400

* Merged commit includes the following changes:
263863588  by yongzhe:

    Fix a bug that the SetExternalContext for EdgeTPU wasn't called when initializing LSTD client.

--
263370193  by yongzhe:

    Internal change.

--

PiperOrigin-RevId: 263863588

* Revert changes in seq_dataset_builder_test.py

PiperOrigin-RevId: 264244022

Small tweak to the range based on seeing values
slightly over 29 and slightly under 28. This avoids
false positives that even if valid are not actionable.

PiperOrigin-RevId: 264196955

PiperOrigin-RevId: 264193390

PiperOrigin-RevId: 264174941

PiperOrigin-RevId: 264174913

PiperOrigin-RevId: 263910123

PiperOrigin-RevId: 263879950

PiperOrigin-RevId: 263874363

PiperOrigin-RevId: 263863438

PiperOrigin-RevId: 263858817

PiperOrigin-RevId: 263854996

PiperOrigin-RevId: 263837214

Also add `worker_hosts` and `task_index` flags.  These flags enable running the
model over multiple hosts by passing the cluster information via command line.

Setting `TF_CONFIG` will continue to work.

PiperOrigin-RevId: 263825245

PiperOrigin-RevId: 263812316

PiperOrigin-RevId: 263669433