new parameter

Author: Adoni

SinkhornGAN.ipynb

@@ -29,14 +29,17 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 24,
+   "execution_count": 26,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Use cuda: False\n"
+      "Use cuda: False\n",
+      "hidden_size: 500\n",
+      "z_size: 2\n",
+      "batch_size: 200\n"
      ]
     }
    ],
@@ -50,12 +53,19 @@
     "use_cuda=torch.cuda.is_available()\n",
     "model_name='DC'\n",
     "model_name='MLP'\n",
-    "print('Use cuda: %r'%use_cuda)"
+    "if model_name=='DC':\n",
+    "    hidden_size=64\n",
+    "if model_name=='MLP':\n",
+    "    hidden_size=500\n",
+    "print('Use cuda: %r'%use_cuda)\n",
+    "print('hidden_size: %d'%hidden_size)\n",
+    "print('z_size: %d'%z_size)\n",
+    "print('batch_size: %d'%batch_size)"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 25,
+   "execution_count": 27,
    "metadata": {
     "collapsed": true
    },
@@ -66,7 +76,6 @@
     "    img_size=32\n",
     "    image_chanel = 1\n",
     "    epsilon=1.0\n",
-    "    hidden_size=256\n",
     "    root = './data/mnist/'\n",
     "    download = True\n",
     "    trans = transforms.Compose([\n",

SinkhornGAN.py

@@ -23,7 +23,7 @@
 import os
 
 
-# In[24]:
+# In[26]:
 
 
 z_size=2
@@ -35,18 +35,24 @@
 use_cuda=torch.cuda.is_available()
 model_name='DC'
 model_name='MLP'
+if model_name=='DC':
+    hidden_size=64
+if model_name=='MLP':
+    hidden_size=500
 print('Use cuda: %r'%use_cuda)
+print('hidden_size: %d'%hidden_size)
+print('z_size: %d'%z_size)
+print('batch_size: %d'%batch_size)
 
 
-# In[25]:
+# In[27]:
 
 
 if dataset_name == 'MNIST':
     total_epoch=50000
     img_size=32
     image_chanel = 1
     epsilon=1.0
-    hidden_size=256
     root = './data/mnist/'
     download = True
     trans = transforms.Compose([