Module keras.distribute

Keras' Distribution Strategy library.

Expand source code
# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================
"""Keras' Distribution Strategy library."""

# pylint: disable=unused-import
from keras.distribute import sidecar_evaluator

Sub-modules

keras.distribute.dataset_creator_model_fit_test_base

Tests for DatasetCreator with Model.fit across usages and strategies.

keras.distribute.distribute_coordinator_utils

Utilities related to distribute coordinator …

keras.distribute.distribute_strategy_test

Tests for tf.keras models using tf.distribute.Strategy.

keras.distribute.distributed_file_utils

Utilities that help manage directory path in distributed settings …

keras.distribute.distributed_training_utils

Utilities related to distributed training.

keras.distribute.distributed_training_utils_v1

Utilities related to distributed training.

keras.distribute.keras_correctness_test_base

Correctness tests for tf.keras using DistributionStrategy.

keras.distribute.keras_dnn_correctness_test

Correctness tests for tf.keras DNN model using DistributionStrategy.

keras.distribute.keras_embedding_model_correctness_test

Correctness test for tf.keras Embedding models using DistributionStrategy.

keras.distribute.keras_image_model_correctness_test

Correctness tests for tf.keras CNN models using DistributionStrategy.

keras.distribute.keras_rnn_model_correctness_test

Correctness tests for tf.keras RNN models using DistributionStrategy.

keras.distribute.keras_stateful_lstm_model_correctness_test

Tests for stateful tf.keras LSTM models using DistributionStrategy.

keras.distribute.keras_utils_test

Tests for tf.keras models with callbacks, checkpointing with dist strategy.

keras.distribute.model_collection_base

A base class to provide a model and corresponding input data for testing.

keras.distribute.model_combinations

Strategy and optimizer combinations for combinations.combine().

keras.distribute.multi_worker_testing_utils

Utilities for testing multi-worker distribution strategies with Keras.

keras.distribute.optimizer_combinations

Strategy and optimizer combinations for combinations.combine().

keras.distribute.saved_model_test_base

Base class for testing saving/loading with DS.

keras.distribute.sidecar_evaluator

Python module for evaluation loop.

keras.distribute.simple_models

A simple functional keras model with one layer.

keras.distribute.strategy_combinations

Strategy combinations for combinations.combine().

keras.distribute.test_example

A simple network to use in tests and examples.

keras.distribute.worker_training_state

Training state management.