Tutorial 9: Fixing callback for PL v1.8

Author: phlippe

docs/tutorial_notebooks/tutorial9/AE_CIFAR10.ipynb

@@ -2106,7 +2106,7 @@
     "        self.input_imgs = input_imgs # Images to reconstruct during training\n",
     "        self.every_n_epochs = every_n_epochs # Only save those images every N epochs (otherwise tensorboard gets quite large)\n",
     "        \n",
-    "    def on_epoch_end(self, trainer, pl_module):\n",
+    "    def on_train_epoch_end(self, trainer, pl_module):\n",
     "        if trainer.current_epoch % self.every_n_epochs == 0:\n",
     "            # Reconstruct images\n",
     "            input_imgs = self.input_imgs.to(pl_module.device)\n",