.. | ||
__init__.py | ||
BUILD | ||
checkpointing_test.py | ||
collective_all_reduce_strategy_test.py | ||
ctl_correctness_test.py | ||
custom_training_loop_metrics_test.py | ||
custom_training_loop_models_test.py | ||
custom_training_loop_optimizer_test.py | ||
distribute_strategy_test.py | ||
distributed_file_utils_test.py | ||
distributed_file_utils.py | ||
distributed_training_utils_test.py | ||
distributed_training_utils_v1.py | ||
distributed_training_utils.py | ||
keras_correctness_test_base.py | ||
keras_dnn_correctness_test.py | ||
keras_embedding_model_correctness_test.py | ||
keras_image_model_correctness_test.py | ||
keras_metrics_test.py | ||
keras_models_test.py | ||
keras_optimizer_v2_test.py | ||
keras_premade_models_test.py | ||
keras_rnn_model_correctness_test.py | ||
keras_save_load_test.py | ||
keras_stateful_lstm_model_correctness_test.py | ||
keras_utils_test.py | ||
minimize_loss_test.py | ||
mirrored_strategy_test.py | ||
mirrored_variable_test.py | ||
model_collection_base.py | ||
model_combinations.py | ||
multi_worker_callback_tf2_test.py | ||
multi_worker_test.py | ||
multi_worker_testing_utils.py | ||
optimizer_combinations.py | ||
parameter_server_training_test.py | ||
README.md | ||
saved_model_mixed_api_test.py | ||
saved_model_save_load_test.py | ||
saved_model_test_base.py | ||
sharded_variable_test.py | ||
sidecar_evaluator_test.py | ||
sidecar_evaluator.py | ||
simple_models.py | ||
strategy_combinations.py | ||
test_example.py | ||
tpu_strategy_test_utils.py | ||
worker_training_state_test.py | ||
worker_training_state.py |
Keras with Distribution Strategy Tests
This directory contains unit tests that combine Keras library with Distribution Training. Tests that use a custom training loop instead of Keras compile/fit should be placed under python/distribute directory instead.