Python clone_model_on_replicas Exemples

Exemple #1

def _build_model(strategy, model, mode, inputs, targets=None):
  if model._compile_distribution:
    distributed_training_utils.clone_model_on_replicas(
        model, strategy, mode, inputs=inputs, targets=targets)
  else:
    distributed_training_utils._build_distributed_network(
        model, strategy, mode, inputs, targets)

Exemple #2

def _build_model(strategy, model, mode, inputs, targets=None):
  if model._compile_distribution:
    distributed_training_utils.clone_model_on_replicas(
        model, strategy, mode, inputs=inputs, targets=targets)
  else:
    distributed_training_utils._build_distributed_network(
        model, strategy, mode, inputs, targets)