diff --git a/site/en/tutorials/images/transfer_learning.ipynb b/site/en/tutorials/images/transfer_learning.ipynb index 57dbbfbcbb..172bb2700b 100644 --- a/site/en/tutorials/images/transfer_learning.ipynb +++ b/site/en/tutorials/images/transfer_learning.ipynb @@ -1049,9 +1049,6 @@ "# Retrieve a batch of images from the test set\n", "image_batch, label_batch = test_dataset.as_numpy_iterator().next()\n", "predictions = model.predict_on_batch(image_batch).flatten()\n", - "\n", - "# Apply a sigmoid since our model returns logits\n", - "predictions = tf.nn.sigmoid(predictions)\n", "predictions = tf.where(predictions < 0.5, 0, 1)\n", "\n", "print('Predictions:\\n', predictions.numpy())\n",