fix long time running

Signed-off-by: YunLiu <[email protected]>

Author: KumoLiu

generation/2d_diffusion_autoencoder/2d_diffusion_autoencoder_tutorial.ipynb

@@ -474,9 +474,7 @@
     }
    ],
    "source": [
-    "max_epochs = (\n",
-    "    1000  # training for longer (1e4 ~ 3h) helps a lot with reconstruction quality, even if the loss is already low\n",
-    ")\n",
+    "max_epochs = 1000  # training for longer (1e4 ~ 3h) helps a lot with reconstruction quality, even if the loss is already low\n",
     "val_interval = 100\n",
     "print_interval = 50\n",
     "iter_loss_list, val_iter_loss_list = [], []\n",