Skip to content

Commit 2ee33e1

Browse files
authored
Fixed training_step t_span
1 parent a71cdcf commit 2ee33e1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tutorials/00_quickstart.ipynb

+1-1
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,7 @@
237237
" \n",
238238
" def training_step(self, batch, batch_idx):\n",
239239
" x, y = batch \n",
240-
" t_eval, y_hat = self.model(x, t_span)\n",
240+
" t_eval, y_hat = self.model(x, self.t_span)\n",
241241
" y_hat = y_hat[-1] # select last point of solution trajectory\n",
242242
" loss = nn.CrossEntropyLoss()(y_hat, y)\n",
243243
" return {'loss': loss} \n",

0 commit comments

Comments
 (0)