[script, trainer] typos ...

bob/learn/pytorch/scripts/train_conditionalgan_multipie.py

@@ -120,8 +120,8 @@ def main(user_input=None):
   
   # WARNING with the transforms ... act on labels too, at some point, I may have to write my own
   # Also, in 'ToTensor', there is a reshape performed from: HxWxC to CxHxW
-  face_dataset = MultiPIEDataset(#root_dir='/Users/guillaumeheusch/work/idiap/data/multipie-cropped-64x64', 
-                                  root_dir='idiap/temp/heusch/data/multipie-cropped-64x64', 
+  face_dataset = MultiPIEDataset(root_dir='/Users/guillaumeheusch/work/idiap/data/multipie-cropped-64x64', 
+                                 # root_dir='idiap/temp/heusch/data/multipie-cropped-64x64', 
                                  frontal_only=False, 
                                  #transform=None
                                  transform=transforms.Compose([

bob/learn/pytorch/trainers/ConditionalGANTrainer2.py

@@ -135,7 +135,7 @@ class ConditionalGANTrainer2(object):
         batch_size = len(real_images)
        
         # create the Tensors with the right batch size
-        noise = torch.FloatTensor(batch_size, noise_dim, 1, 1).normal_(0, 1)
+        noise = torch.FloatTensor(batch_size, self.noise_dim, 1, 1).normal_(0, 1)
         label = torch.FloatTensor(batch_size)
         
         if self.use_gpu: