def save_model(self, epoch=None):
if epoch is not None:
fname = self.conf.save_model.replace('%e', str(epoch).zfill(5))
else:
fname = self.conf.save_model.replace('%e', 'final')
if self.conf.verbosity > 1:
print "Saving model to", fname
np.savez(fname, *get_all_param_values(self.autoencoder))
评论列表
文章目录