[trainer] small synchro

bob/learn/pytorch/trainers/ConditionalGANTrainer.py

@@ -119,14 +119,15 @@ class ConditionalGANTrainer(object):
      
         start = time.time()
 
-        # ===========================================================
-        # (1) Update D network: maximize log(D(x)) + log(1 - D(G(z)))
-        # ===========================================================
+        # =============
+        # DISCRIMINATOR 
+        # =============
       
         # train with real
         self.netD.zero_grad()
         real_images = data['image']
         poses = data['pose']
+        
         image_size = real_images[1].size()
 
         # WARNING: the last batch could be smaller than the provided size