more comments

cgopen · Jun 17, 2018 · d2ead07 · d2ead07
1 parent 8b3dd65
commit d2ead07
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/trainer.py b/src/trainer.py
@@ -212,6 +212,7 @@ def build_model(self):
         # For discriminator
         fake_rotations, fake_shapes = [], []
         # Start loop
+        # 85D
         theta_prev = self.load_mean_param()
 
         # For visualizations
@@ -221,6 +222,7 @@ def build_model(self):
         self.all_delta_thetas = []
         self.all_theta_prev = []
 
+        # Main IEF loop
         for i in np.arange(self.num_stage):
             print('Iteration %d' % i)
             # ---- Compute outputs