experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/python/keras/distribute
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
..
__init__.py
Add __init__.py in keras/distribute directory.
2019-04-09 21:47:40 -07:00
BUILD
Fork keras related mirror_variable_test to keras/distribute.
2020-06-19 10:57:00 -07:00
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_test.py
Add support for user-defined TensorBoard summaries in Layers/Models in single
2019-08-07 13:54:40 -07:00
distributed_training_utils.py
Drop experimental and v2 qualifiers from Strategy experimental_run_v2 method.
2020-03-12 10:26:26 -07:00
keras_correctness_test_base.py
Remove passing experimental_run_tf_function in most tests.
2020-02-27 13:34:09 -08:00
keras_dnn_correctness_test.py
Remove passing experimental_run_tf_function in most tests.
2020-02-27 13:34:09 -08:00
keras_embedding_model_correctness_test.py
Remove passing experimental_run_tf_function in most tests.
2020-02-27 13:34:09 -08:00
keras_image_model_correctness_test.py
Remove passing experimental_run_tf_function in most tests.
2020-02-27 13:34:09 -08:00
keras_optimizer_v2_test.py
Remove passing experimental_run_tf_function in most tests.
2020-02-27 13:34:09 -08:00
keras_premade_models_test.py
Remove passing experimental_run_tf_function in most tests.
2020-02-27 13:34:09 -08:00
keras_rnn_model_correctness_test.py
Remove passing experimental_run_tf_function in most tests.
2020-02-27 13:34:09 -08:00
keras_stateful_lstm_model_correctness_test.py
Remove passing experimental_run_tf_function in most tests.
2020-02-27 13:34:09 -08:00
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
Make the check for whether it is multi_worker_mode not rely on TF_CONFIG environment variable, but delegate to strategy extended objects.
2019-08-22 12:16:41 -07:00
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
README.md
Add Readme.md for keras/distribute.
2020-02-07 16:17:20 -08:00
tpu_strategy_test_utils.py
Add distribution strategy tests.
2020-04-23 11:14:01 -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
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

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.