diff options
| author |  Jianwei Xie <xiejw@google.com> | 2018-03-29 10:50:46 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-29 10:52:55 -0700 |
| commit | 63dffd5a3bc4e94e74cb140cbf7a68e0e5644ad6 (patch) | |
| tree | 087e334cd7e71d44cdcd1ce5ebab483768fcf474 /tensorflow/contrib/estimator | |
| parent | 9fbb5b3b8fef1caa2ee2ca4a0f8dde900d1f2aa5 (diff) | |
Automated g4 rollback of changelist 190858242
PiperOrigin-RevId: 190953197
Diffstat (limited to 'tensorflow/contrib/estimator')
| -rw-r--r-- | tensorflow/contrib/estimator/python/estimator/replicate_model_fn.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/estimator/python/estimator/replicate_model_fn.py b/tensorflow/contrib/estimator/python/estimator/replicate_model_fn.py index e0fae2c992..fa2697800e 100644 --- a/tensorflow/contrib/estimator/python/estimator/replicate_model_fn.py +++ b/tensorflow/contrib/estimator/python/estimator/replicate_model_fn.py @@ -136,7 +136,7 @@ def replicate_model_fn(model_fn, the train_op argument of `EstimatorSpec`. loss_reduction: controls whether losses are summed or averaged. devices: Optional list of devices to replicate the model across. This - argument can be used to replice only on the subset of available GPUs. + argument can be used to replicate only on the subset of available GPUs. If `None`, then all available GPUs are going to be used for replication. If no GPUs are available, then the model is going to be placed on the CPU. |