diff --git a/DCGAN-PyTorch/01-DCGAN-PL.ipynb b/DCGAN-PyTorch/01-DCGAN-PL.ipynb
index fdb048ab83c82dabef2958a3901c8a7ef679ff65..a0973d1262fef6a52b6e176a0affd7823b6f5cfc 100644
--- a/DCGAN-PyTorch/01-DCGAN-PL.ipynb
+++ b/DCGAN-PyTorch/01-DCGAN-PL.ipynb
@@ -1,7 +1,6 @@
 {
  "cells": [
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -83,8 +82,8 @@
     "generator_class     = 'Generator_1'\n",
     "discriminator_class = 'Discriminator_1'    \n",
     "    \n",
-    "scale               = .05\n",
-    "epochs              = 15\n",
+    "scale               = .1\n",
+    "epochs              = 10\n",
     "batch_size          = 32\n",
     "num_img             = 36\n",
     "fit_verbosity       = 2\n",
@@ -94,7 +93,6 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -102,7 +100,6 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -124,7 +121,6 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -145,7 +141,6 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -159,7 +154,6 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -199,7 +193,6 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -222,7 +215,6 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -291,7 +283,6 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -317,9 +308,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "pytorch-gpu-1.13.0_py3.10.8",
    "language": "python",
-   "name": "python3"
+   "name": "module-conda-env-pytorch-gpu-1.13.0_py3.10.8"
   },
   "language_info": {
    "codemirror_mode": {
@@ -331,7 +322,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.2"
+   "version": "3.10.8"
   },
   "vscode": {
    "interpreter": {
diff --git a/DCGAN-PyTorch/modules/GAN.py b/DCGAN-PyTorch/modules/GAN.py
index 8a1f3ad4b3042d1b70376e7b99f6f163d1247019..a97065c36b1aaa0f37e769474f5147220f35ae01 100644
--- a/DCGAN-PyTorch/modules/GAN.py
+++ b/DCGAN-PyTorch/modules/GAN.py
@@ -157,6 +157,7 @@ class GAN(LightningModule):
         #
         if(self.current_epoch==1):
             sampleImg=torch.rand((1,28,28,1))
+            sampleImg=sampleImg.type_as(self.generator.model[0].weight)
             self.logger.experiment.add_graph(self.discriminator,sampleImg)
 
         # ---- Log d_loss/epoch