def save(self, folder, epoch, batch):
print('-'*5 , 'saving model')
serializers.save_hdf5('{}/network_epoch{}_batch{}.model'.format(folder, epoch, batch), self.Networks[0])
print('-'*5 , 'saving optimizer')
serializers.save_hdf5('{}/network_epoch{}_batch{}.state'.format(folder, epoch, batch), self.Optimizer)
return
评论列表
文章目录