From 82ea8793f980bdce8e827cf3cd6b09be8d8435b4 Mon Sep 17 00:00:00 2001
From: james47 <aiyujie2006@gmail.com>
Date: Tue, 5 Nov 2024 19:44:03 -0800
Subject: [PATCH] Update reset_states() -> reset_state() in Custom Training
 Loops Tutorials

---
 site/en/tutorials/distribute/custom_training.ipynb | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/site/en/tutorials/distribute/custom_training.ipynb b/site/en/tutorials/distribute/custom_training.ipynb
index d14b0ac003c..9d1057e7ca9 100644
--- a/site/en/tutorials/distribute/custom_training.ipynb
+++ b/site/en/tutorials/distribute/custom_training.ipynb
@@ -514,9 +514,9 @@
         "                         train_accuracy.result() * 100, test_loss.result(),\n",
         "                         test_accuracy.result() * 100))\n",
         "\n",
-        "  test_loss.reset_states()\n",
-        "  train_accuracy.reset_states()\n",
-        "  test_accuracy.reset_states()"
+        "  test_loss.reset_state()\n",
+        "  train_accuracy.reset_state()\n",
+        "  test_accuracy.reset_state()"
       ]
     },
     {
@@ -633,7 +633,7 @@
         "\n",
         "  template = (\"Epoch {}, Loss: {}, Accuracy: {}\")\n",
         "  print(template.format(epoch + 1, average_train_loss, train_accuracy.result() * 100))\n",
-        "  train_accuracy.reset_states()"
+        "  train_accuracy.reset_state()"
       ]
     },
     {
@@ -672,7 +672,7 @@
         "  template = (\"Epoch {}, Loss: {}, Accuracy: {}\")\n",
         "  print(template.format(epoch + 1, train_loss, train_accuracy.result() * 100))\n",
         "\n",
-        "  train_accuracy.reset_states()"
+        "  train_accuracy.reset_state()"
       ]
     },
     {