Skip to content

Commit 14820d1

Browse files
authored
Update legacy API usage in get_started.ipynb (#6932)
## Motivation for features / changes As title. Current code no longer runs in the newer TF version (2.17.0 on Colab) with the following error / warning: ``` AttributeError Traceback (most recent call last) [<ipython-input-19-f1056f2ad84b>](https://localhost:8080/#) in <cell line: 5>() 24 25 # Reset metrics every epoch ---> 26 train_loss.reset_states() 27 test_loss.reset_states() 28 train_accuracy.reset_states() AttributeError: 'Mean' object has no attribute 'reset_states' ``` ``` /usr/local/lib/python3.10/dist-packages/keras/src/layers/reshaping/flatten.py:37: UserWarning: Do not pass an `input_shape`/`input_dim` argument to a layer. When using Sequential models, prefer using an `Input(shape)` object as the first layer in the model instead. super().__init__(**kwargs) ``` ## Technical description of changes N/A ## Screenshots of UI changes (or N/A) N/A ## Detailed steps to verify changes work correctly (as executed by you) Run on Colab ## Alternate designs / implementations considered (or N/A) N/A
1 parent 49dda43 commit 14820d1

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

docs/get_started.ipynb

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,8 @@
135135
"\n",
136136
"def create_model():\n",
137137
" return tf.keras.models.Sequential([\n",
138-
" tf.keras.layers.Flatten(input_shape=(28, 28), name='layers_flatten'),\n",
138+
" tf.keras.layers.Input(shape=(28, 28), name='layers_input'),\n",
139+
" tf.keras.layers.Flatten(name='layers_flatten'),\n",
139140
" tf.keras.layers.Dense(512, activation='relu', name='layers_dense'),\n",
140141
" tf.keras.layers.Dropout(0.2, name='layers_dropout'),\n",
141142
" tf.keras.layers.Dense(10, activation='softmax', name='layers_dense_2')\n",
@@ -452,10 +453,10 @@
452453
" test_accuracy.result()*100))\n",
453454
"\n",
454455
" # Reset metrics every epoch\n",
455-
" train_loss.reset_states()\n",
456-
" test_loss.reset_states()\n",
457-
" train_accuracy.reset_states()\n",
458-
" test_accuracy.reset_states()"
456+
" train_loss.reset_state()\n",
457+
" test_loss.reset_state()\n",
458+
" train_accuracy.reset_state()\n",
459+
" test_accuracy.reset_state()"
459460
]
460461
},
461462
{

0 commit comments

Comments
 (0)