Skip to content

Commit

Permalink
Update docs/tutorials/05_torch_connector.ipynb
Browse files Browse the repository at this point in the history
Co-authored-by: Anton Dekusar <[email protected]>
  • Loading branch information
FelixLehn and adekusar-drl authored Oct 31, 2023
1 parent f7a0fc3 commit 399b573
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/tutorials/05_torch_connector.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -321,7 +321,7 @@
],
"source": [
"# Evaluate model and compute accuracy\n",
"model1.eval() # set model to evaluation mode\n",
"model1.eval()\n",
"y_predict = []\n",
"for x, y_target in zip(X, y):\n",
" output = model1(Tensor(x))\n",
Expand Down

0 comments on commit 399b573

Please sign in to comment.