formatting

Author: bstriner

recitation-10/Pytorch GANs.ipynb

@@ -68,7 +68,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "mnist_gan.main(['--generator-frequency=5', '--save-directory=output/mnist_gan/frequency-5', '--epochs=100'])"
+    "mnist_gan.main([])"
    ]
   },
   {
@@ -118,7 +118,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "mnist_wgangp.main()"
+    "mnist_wgangp.main([])"
    ]
   },
   {
@@ -143,7 +143,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "cifar10_wgangp.main()"
+    "cifar10_wgangp.main([])"
    ]
   },
   {
@@ -168,7 +168,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "mnist_cwgangp.main()"
+    "mnist_cwgangp.main([])"
    ]
   },
   {

recitation-10/mnist_cwgangp.py

@@ -220,7 +220,7 @@ def main(argv):
     parser = argparse.ArgumentParser(description='PyTorch GAN Example')
 
     # Output directory
-    parser.add_argument('--save-directory', type=str, default='output/mnist_cwgangp/v2', help='output directory')
+    parser.add_argument('--save-directory', type=str, default='output/mnist_cwgangp/v1', help='output directory')
 
     # Configuration
     parser.add_argument('--batch-size', type=int, default=128, metavar='N', help='batch size')

recitation-10/mnist_gan.py

@@ -221,7 +221,7 @@ def train_generator(self):
         torch.randn(*latent.size(), out=latent)
         latent = Variable(latent)
         # Calculate yfake
-        y = Variable(torch.rand(latent.size(0), out=latent.data.new())*10).long()
+        y = Variable(torch.rand(latent.size(0), out=latent.data.new()) * 10).long()
         yfake = self.trainer.model.y_fake(latent, y)
         # Calculate loss
         loss = self.criterion(yfake)

recitation-10/mnist_wgangp.py

@@ -103,7 +103,7 @@ def main(argv):
     parser = argparse.ArgumentParser(description='PyTorch GAN Example')
 
     # Output directory
-    parser.add_argument('--save-directory', type=str, default='output/mnist_wgangp/v8', help='output directory')
+    parser.add_argument('--save-directory', type=str, default='output/mnist_wgangp/v1', help='output directory')
 
     # Configuration
     parser.add_argument('--batch-size', type=int, default=128, metavar='N', help='batch size')