Skip to content

Commit 8041008

Browse files
committed
Minor fixes
1 parent 660498d commit 8041008

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

chapter03_introduction-to-keras-and-tf.ipynb

+1-1
Original file line numberDiff line numberDiff line change
@@ -590,7 +590,7 @@
590590
"def training_step(inputs, targets):\n",
591591
" with tf.GradientTape() as tape:\n",
592592
" predictions = model(inputs)\n",
593-
" loss = square_loss(predictions, targets)\n",
593+
" loss = square_loss(targets, predictions)\n",
594594
" grad_loss_wrt_W, grad_loss_wrt_b = tape.gradient(loss, [W, b])\n",
595595
" W.assign_sub(grad_loss_wrt_W * learning_rate)\n",
596596
" b.assign_sub(grad_loss_wrt_b * learning_rate)\n",

chapter07_working-with-keras.ipynb

+3-3
Original file line numberDiff line numberDiff line change
@@ -1329,8 +1329,8 @@
13291329
" with tf.GradientTape() as tape:\n",
13301330
" predictions = self(inputs, training=True)\n",
13311331
" loss = loss_fn(targets, predictions)\n",
1332-
" gradients = tape.gradient(loss, model.trainable_weights)\n",
1333-
" optimizer.apply_gradients(zip(gradients, model.trainable_weights))\n",
1332+
" gradients = tape.gradient(loss, self.trainable_weights)\n",
1333+
" self.optimizer.apply_gradients(zip(gradients, self.trainable_weights))\n",
13341334
"\n",
13351335
" loss_tracker.update_state(loss)\n",
13361336
" return {\"loss\": loss_tracker.result()}\n",
@@ -1436,4 +1436,4 @@
14361436
},
14371437
"nbformat": 4,
14381438
"nbformat_minor": 0
1439-
}
1439+
}

0 commit comments

Comments
 (0)