Internal change

PiperOrigin-RevId: 422612138

Internal change
PiperOrigin-RevId: 422612138
4242b549 · A. Unique TensorFlower · 43b55bf4 · 4242b549 · 4242b549 · 4242b549
Commit 4242b549 authored Jan 18, 2022 by A. Unique TensorFlower
20 changed files
--- a/official/modeling/fast_training/experimental/tf2_utils_2x_wide_test.py
+++ b/official/modeling/fast_training/experimental/tf2_utils_2x_wide_test.py
--- a/official/modeling/fast_training/progressive/policies.py
+++ b/official/modeling/fast_training/progressive/policies.py
--- a/official/modeling/fast_training/progressive/train.py
+++ b/official/modeling/fast_training/progressive/train.py
--- a/official/modeling/fast_training/progressive/train_lib.py
+++ b/official/modeling/fast_training/progressive/train_lib.py
--- a/official/modeling/fast_training/progressive/train_lib_test.py
+++ b/official/modeling/fast_training/progressive/train_lib_test.py
--- a/official/modeling/fast_training/progressive/trainer.py
+++ b/official/modeling/fast_training/progressive/trainer.py
--- a/official/modeling/fast_training/progressive/trainer_test.py
+++ b/official/modeling/fast_training/progressive/trainer_test.py
--- a/official/modeling/fast_training/progressive/utils.py
+++ b/official/modeling/fast_training/progressive/utils.py
--- a/official/modeling/grad_utils.py
+++ b/official/modeling/grad_utils.py
--- a/official/modeling/grad_utils_test.py
+++ b/official/modeling/grad_utils_test.py
--- a/official/modeling/hyperparams/__init__.py
+++ b/official/modeling/hyperparams/__init__.py
--- a/official/modeling/hyperparams/base_config.py
+++ b/official/modeling/hyperparams/base_config.py
--- a/official/modeling/hyperparams/base_config_test.py
+++ b/official/modeling/hyperparams/base_config_test.py
--- a/official/modeling/hyperparams/oneof.py
+++ b/official/modeling/hyperparams/oneof.py
--- a/official/modeling/hyperparams/oneof_test.py
+++ b/official/modeling/hyperparams/oneof_test.py
--- a/official/modeling/hyperparams/params_dict.py
+++ b/official/modeling/hyperparams/params_dict.py
--- a/official/modeling/hyperparams/params_dict_test.py
+++ b/official/modeling/hyperparams/params_dict_test.py
--- a/official/modeling/multitask/__init__.py
+++ b/official/modeling/multitask/__init__.py
--- a/official/modeling/multitask/base_model.py
+++ b/official/modeling/multitask/base_model.py
--- a/official/modeling/multitask/base_trainer.py
+++ b/official/modeling/multitask/base_trainer.py