fixed a bug in not saving the very last iteration

Author: taesungp

experiments/church_launcher.py

@@ -20,12 +20,6 @@ def options(self):
                 name="church_default",
                 patch_use_aggregation=False,
             ),
-            opt.specify(
-                name="church_conservative",
-                patch_use_aggregation=False,
-                lambda_patch_R1=10.0,
-                resume_iter="24950k",
-            ),
         ]
 
     def train_options(self):
@@ -45,7 +39,7 @@ def test_options(self):
         return [opt.tag("fig4").specify(
             num_gpus=1,
             batch_size=1,
-            dataroot="~/datasets/testphotos/images_used_in_swapping_autoencoder_arxiv/church/fig4",
+            dataroot="./testphotos/church/fig4",
             dataset_mode="imagefolder",
             preprocess="scale_shortside",  # For testing, scale but don't crop
             evaluation_metrics="structure_style_grid_generation",

train.py

@@ -51,4 +51,5 @@
 
         iter_counter.record_one_iteration()
 
+optimizer.save(iter_counter.steps_so_far)
 print('Training finished.')

util/iter_counter.py

@@ -54,8 +54,7 @@ def record_one_iteration(self):
         self.steps_so_far += self.batch_size
 
     def needs_saving(self):
-        return self.completed_training() or \
-            (self.steps_so_far % self.opt.save_freq) < self.batch_size
+        return (self.steps_so_far % self.opt.save_freq) < self.batch_size
 
     def needs_evaluation(self):
         return (self.steps_so_far >= self.opt.evaluation_freq) and \