commit 856698be1fe9c624310fbcc131c002b15b3d0b12
author Zhenyu Tan <tanzheny@google.com> Wed Nov 18 08:43:18 2020 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> Wed Nov 18 08:48:03 2020 -0800
tree 2f4aa023b60b921c0f2abab3e5405ab4fd1babe9
parent 402d28705e426fea7aad6bbbe405a11daa6b6cd5

Update worker_training_state comment.

PiperOrigin-RevId: 343086387
Change-Id: I703e22ff895b6f2a41dedfac9ffd9a27f8c64491

tensorflow/python/keras/engine/training.py

diff --git a/tensorflow/python/keras/engine/training.py b/tensorflow/python/keras/engine/training.py
index a644b95..55f71e3 100644
--- a/tensorflow/python/keras/engine/training.py
+++ b/tensorflow/python/keras/engine/training.py
@@ -2584,7 +2584,7 @@
   def _maybe_load_initial_epoch_from_ckpt(self, initial_epoch):
     """Maybe load initial epoch from ckpt considering possible worker recovery.
 
-    Refer to tensorflow/python/keras/distribute/multi_worker_training_state.py
+    Refer to tensorflow/python/keras/distribute/worker_training_state.py
     for more information.
 
     Arguments:

tensorflow/python/keras/engine/training_v1.py

diff --git a/tensorflow/python/keras/engine/training_v1.py b/tensorflow/python/keras/engine/training_v1.py
index 6617c2d..c4027c0 100644
--- a/tensorflow/python/keras/engine/training_v1.py
+++ b/tensorflow/python/keras/engine/training_v1.py
@@ -2753,7 +2753,7 @@
   def _maybe_load_initial_epoch_from_ckpt(self, initial_epoch, mode):
     """Maybe load initial epoch from ckpt considering possible worker recovery.
 
-    Refer to tensorflow/python/keras/distribute/multi_worker_training_state.py
+    Refer to tensorflow/python/keras/distribute/worker_training_state.py
     for more information.
 
     Arguments: