def do(self, callback_name, *args): """Pickle the main loop object to the disk. If `*args` contain an argument from user, it is treated as saving path to be used instead of the one given at the construction stage. """ from_main_loop, from_user = self.parse_args(callback_name, args) try: path = self.path if len(from_user): path, = from_user # already_saved_to = self.main_loop.log.current_row.get(SAVED_TO, ()) # self.main_loop.log.current_row[SAVED_TO] = ( # already_saved_to + (path,)) # secure_pickle_dump(self.main_loop, path) filenames = self.save_separately_filenames(path) for attribute in self.save_separately: p = getattr(self.main_loop, attribute) if p: secure_pickle_dump(p, filenames[attribute]) else: print("Empty %s",attribute) generate_samples(self.main_loop.model, self.save_subdir, self.image_size) if os.path.exists(self.epoch_src): epoch_dst = "{0}/epoch-{1:03d}.png".format(self.save_subdir, self.iteration) self.iteration = self.iteration + 1 shutil.copy2(self.epoch_src, epoch_dst) os.system("convert -delay 5 -loop 1 {0}/epoch-*.png {0}/training.gif".format(self.save_subdir)) except Exception: self.main_loop.log.current_row[SAVED_TO] = None raise
def do(self, callback_name, *args): """Pickle the main loop object to the disk. If `*args` contain an argument from user, it is treated as saving path to be used instead of the one given at the construction stage. """ from_main_loop, from_user = self.parse_args(callback_name, args) try: path = self.path if len(from_user): path, = from_user # already_saved_to = self.main_loop.log.current_row.get(SAVED_TO, ()) # self.main_loop.log.current_row[SAVED_TO] = ( # already_saved_to + (path,)) # secure_pickle_dump(self.main_loop, path) filenames = self.save_separately_filenames(path) for attribute in self.save_separately: p = getattr(self.main_loop, attribute) if p: secure_pickle_dump(p, filenames[attribute]) else: print("Empty %s", attribute) generate_samples(self.main_loop, self.image_size) except Exception: self.main_loop.log.current_row[SAVED_TO] = None raise
def do(self, callback_name, *args): """Sample the model and save images to disk """ generate_samples(self.main_loop.model, self.save_subdir, self.image_size, self.channels) if os.path.exists(self.epoch_src): epoch_dst = "{0}/epoch-{1:03d}.png".format(self.save_subdir, self.iteration) self.iteration = self.iteration + 1 shutil.copy2(self.epoch_src, epoch_dst) os.system("convert -delay 5 -loop 1 {0}/epoch-*.png {0}/training.gif".format(self.save_subdir))
def do(self, callback_name, *args): """Sample the model and save images to disk """ generate_samples(self.main_loop.model, self.save_subdir, self.image_size, self.channels) if os.path.exists(self.epoch_src): epoch_dst = "{0}/epoch-{1:03d}.png".format(self.save_subdir, self.iteration) self.iteration = self.iteration + 1 shutil.copy2(self.epoch_src, epoch_dst) os.system( "convert -delay 5 -loop 1 {0}/epoch-*.png {0}/training.gif". format(self.save_subdir))