Merge pull request #77 from anchitagarwal/master

data_params fix
ragavvenkatesan · Jun 2, 2017 · e0f883c · e0f883c
2 parents 7a65f3b + b4218c9
commit e0f883c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/yann/special/datasets.py b/yann/special/datasets.py
@@ -595,13 +595,13 @@ def cook_celeba(verbose = 1, **kwargs):
                     "source"             : 'images',
                     "name"               : 'celeba',
                     "location"           : '',
-                    "mini_batch_size"    : 36,
+                    "mini_batch_size"    : 500,
                     "mini_batches_per_batch" : (1, 1, 1),
-                    "batches2train"      : 1,
+                    "batches2train"      : 403,
                     "batches2test"       : 1,
                     "batches2validate"   : 1,
-                    "height"             : 218,
-                    "width"              : 178,
+                    "height"             : 32,
+                    "width"              : 32,
                     "channels"           : 3  }
 
     else: