• Home
Name Date Size #Lines LOC

..--

BUILDD03-May-202430 KiB977925

README.mdD03-May-2024316 75

__init__.pyD03-May-2024941 235

checkpointing_test.pyD03-May-20245 KiB14399

collective_all_reduce_strategy_test.pyD03-May-20243.2 KiB8456

ctl_correctness_test.pyD03-May-202410.1 KiB279223

custom_training_loop_metrics_test.pyD03-May-20244.5 KiB13184

custom_training_loop_models_test.pyD03-May-202417.3 KiB514380

custom_training_loop_optimizer_test.pyD03-May-20244.4 KiB12285

distribute_strategy_test.pyD03-May-2024103.1 KiB2,7142,069

distributed_file_utils.pyD03-May-20245.5 KiB15092

distributed_file_utils_test.pyD03-May-20245.4 KiB13699

distributed_training_utils.pyD03-May-20242.2 KiB6130

distributed_training_utils_test.pyD03-May-20242.1 KiB5934

distributed_training_utils_v1.pyD03-May-202446.3 KiB1,159805

keras_correctness_test_base.pyD03-May-202421.4 KiB631463

keras_dnn_correctness_test.pyD03-May-202413.1 KiB331265

keras_embedding_model_correctness_test.pyD03-May-20246.3 KiB161116

keras_image_model_correctness_test.pyD03-May-20246.5 KiB165129

keras_metrics_test.pyD03-May-20245 KiB13174

keras_models_test.pyD03-May-20242.2 KiB6237

keras_optimizer_v2_test.pyD03-May-20245.2 KiB146100

keras_premade_models_test.pyD03-May-20244.1 KiB10373

keras_rnn_model_correctness_test.pyD03-May-20245.4 KiB137100

keras_save_load_test.pyD03-May-20243.4 KiB7953

keras_stateful_lstm_model_correctness_test.pyD03-May-20244 KiB11277

keras_utils_test.pyD03-May-202420.9 KiB581475

minimize_loss_test.pyD03-May-202422.8 KiB554427

mirrored_strategy_test.pyD03-May-20245.5 KiB147103

mirrored_variable_test.pyD03-May-20244 KiB10871

model_collection_base.pyD03-May-20241.6 KiB4624

model_combinations.pyD03-May-20241.4 KiB3514

multi_worker_callback_tf2_test.pyD03-May-202413.9 KiB368242

multi_worker_test.pyD03-May-20249.2 KiB252188

multi_worker_testing_utils.pyD03-May-20243 KiB7647

optimizer_combinations.pyD03-May-20245 KiB11181

parameter_server_training_test.pyD03-May-202414 KiB365268

saved_model_mixed_api_test.pyD03-May-20243.8 KiB8759

saved_model_save_load_test.pyD03-May-20248.5 KiB186145

saved_model_test_base.pyD03-May-202410.4 KiB290211

sharded_variable_test.pyD03-May-20244 KiB11275

sidecar_evaluator.pyD03-May-20249.9 KiB236169

sidecar_evaluator_test.pyD03-May-20246.9 KiB179117

simple_models.pyD03-May-20244 KiB14791

strategy_combinations.pyD03-May-20242.2 KiB6036

test_example.pyD03-May-20243.5 KiB9960

tpu_strategy_test_utils.pyD03-May-20241.8 KiB5026

worker_training_state.pyD03-May-20245.9 KiB14485

worker_training_state_test.pyD03-May-20242.6 KiB6543

README.md

1# Keras with Distribution Strategy Tests
2
3This directory contains unit tests that combine Keras library with
4[Distribution Training](https://www.tensorflow.org/guide/distributed_training).
5Tests that use a custom training loop instead of Keras compile/fit should be
6placed under python/distribute directory instead.
7