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
withModel.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.