put models in eval mode

Author: csinva

cifar100_dcgan_grayscale/dcgan_cifar100_pretrained_load.ipynb

@@ -22,7 +22,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -31,8 +31,8 @@
     "# load the models\n",
     "from dcgan import Discriminator, Generator\n",
     "\n",
-    "D = Discriminator(ngpu=1)\n",
-    "G = Generator(ngpu=1)\n",
+    "D = Discriminator(ngpu=1).eval()\n",
+    "G = Generator(ngpu=1).eval()\n",
     "\n",
     "# load weights\n",
     "D.load_state_dict(torch.load('weights/netD_epoch_299.pth'))\n",
@@ -142,7 +142,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.5.4"
+   "version": "3.7.0"
   }
  },
  "nbformat": 4,

cifar10_dcgan/dcgan_cifar10_pretrained_load.ipynb

@@ -31,8 +31,8 @@
     "# load the models\n",
     "from dcgan import Discriminator, Generator\n",
     "\n",
-    "D = Discriminator(ngpu=1)\n",
-    "G = Generator(ngpu=1)\n",
+    "D = Discriminator(ngpu=1).eval()\n",
+    "G = Generator(ngpu=1).eval()\n",
     "\n",
     "# load weights\n",
     "D.load_state_dict(torch.load('weights/netD_epoch_199.pth'))\n",
@@ -166,7 +166,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.5.4"
+   "version": "3.7.0"
   }
  },
  "nbformat": 4,

classifiers/cifar/cifar_pretrained_load.ipynb

@@ -30,7 +30,7 @@
    "outputs": [],
    "source": [
     "device = 'cuda' if torch.cuda.is_available() else 'cpu'\n",
-    "net = model.cifar10(128, pretrained='log/cifar10/best-135.pth')  \n",
+    "net = model.cifar10(128, pretrained='log/cifar10/best-135.pth').eval()  \n",
     "net = net.to(device)"
    ]
   }

classifiers/mnist/lenet_pretrained_load.ipynb

@@ -30,7 +30,7 @@
    "outputs": [],
    "source": [
     "device = 'cuda' if torch.cuda.is_available() else 'cpu'\n",
-    "net = LeNet5()\n",
+    "net = LeNet5().eval()\n",
     "net.load_state_dict(torch.load('weights/lenet_epoch=12_test_acc=0.991.pth'))\n",
     "net = net.to(device)"
    ]

mnist_dcgan/dcgan_mnist_pretrained_load.ipynb

@@ -31,8 +31,8 @@
     "# load the models\n",
     "from dcgan import Discriminator, Generator\n",
     "\n",
-    "D = Discriminator(ngpu=1)\n",
-    "G = Generator(ngpu=1)\n",
+    "D = Discriminator(ngpu=1).eval()\n",
+    "G = Generator(ngpu=1).eval()\n",
     "\n",
     "# load weights\n",
     "D.load_state_dict(torch.load('weights/netD_epoch_99.pth'))\n",
@@ -134,7 +134,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.5.4"
+   "version": "3.7.0"
   }
  },
  "nbformat": 4,