bug fix, multiple epochs

data/gan.py

@@ -507,7 +507,8 @@ class Train (GNet):
 
                                         logs.append({"epoch":epoch,"distance":-w_sum/(self.STEPS_PER_EPOCH*2) })
 
-                                        if epoch % self.MAX_EPOCHS == 0:
+                                        # if epoch % self.MAX_EPOCHS == 0:
+                                        if epoch in [5,10,50, self.MAX_EPOCHS] :
                                                 # suffix = "-".join(self.ATTRIBUTES['synthetic']) if isinstance(self.ATTRIBUTES['synthetic'],list) else self.ATTRIBUTES['synthetic']
                                                 suffix = self.get.suffix()
                                                 _name  = os.sep.join([self.train_dir,suffix])