def dump(self, main_loop):
"""Overwrites MainLoopDumpManager.dump()."""
if not os.path.exists(self.folder):
os.mkdir(self.folder)
print ""
logger.info(" Saving model")
start = time.time()
logger.info(" ...saving parameters")
self.dump_parameters(main_loop)
logger.info(" ...saving iteration state")
secure_pickle_dump(main_loop.iteration_state,
self.path_to_iteration_state)
logger.info(" ...saving log")
secure_pickle_dump(main_loop.log, self.path_to_log)
if self.save_accumulators:
logger.info(" ...saving algorithm")
self.dump_accumulators(main_loop)
logger.info(" Model saved, took {} seconds.".format(time.time()-start))
评论列表
文章目录