History

Rick Chao 48064efcfd Skip data loading error in multi_worker_tutorial_test (the test does not aim to cover this). PiperOrigin-RevId: 317797271 Change-Id: I8336d7ffeda0836beef0a2d04e633614a44e7fa4		2020-06-23 15:56:23 -07:00
..
BUILD	Fork keras related mirror_variable_test to keras/distribute.	2020-06-19 10:57:00 -07:00
README.md	…
__init__.py	…
distribute_strategy_test.py	Don't disable all TPU tests just the ones that fail.	2020-05-21 16:02:14 -07:00
distributed_training_utils.py	…
distributed_training_utils_test.py	…
keras_correctness_test_base.py	…
keras_dnn_correctness_test.py	…
keras_embedding_model_correctness_test.py	…
keras_image_model_correctness_test.py	…
keras_optimizer_v2_test.py	…
keras_premade_models_test.py	…
keras_rnn_model_correctness_test.py	…
keras_stateful_lstm_model_correctness_test.py	…
keras_utils_test.py	Enable skipped test as Variable.assign(name=xxx) now works with CentralStorage	2020-05-21 11:02:24 -07:00
mirrored_strategy_test.py	Fork the keras related mirrored_strategy_test to keras/distribute.	2020-06-19 10:27:02 -07:00
mirrored_variable_test.py	Fork keras related mirror_variable_test to keras/distribute.	2020-06-19 10:57:00 -07:00
multi_worker_callback_tf2_test.py	Improve multi_process_runner	2020-06-11 05:20:30 -07:00
multi_worker_test.py	…
multi_worker_testing_utils.py	Update model saving test with MultiWorkerMirroredStrategy.	2020-06-08 02:23:55 -07:00
multi_worker_tutorial_test.py	Skip data loading error in multi_worker_tutorial_test (the test does not aim to cover this).	2020-06-23 15:56:23 -07:00
tpu_strategy_test_utils.py	Add distribution strategy tests.	2020-04-23 11:14:01 -07:00
worker_training_state.py	Add BackupAndRestore callback to help model back up periodically and recover from unexpected failure, e.g. preemption.	2020-05-27 14:19:33 -07:00
worker_training_state_test.py	Add BackupAndRestore callback to help model back up periodically and recover from unexpected failure, e.g. preemption.	2020-05-27 14:19:33 -07:00

README.md

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.