def saveInfo(self, model, optimizer, epoch, outputFolder, saveEach):
if(epoch % saveEach == 0):
if(not os.path.exists(outputFolder)):
os.makedirs(outputFolder)
bname = outputFolder + '/' + model.getName() + '_' + str(epoch)
serializers.save_npz(bname + '.model', model)
serializers.save_npz(bname + '.state', optimizer)
评论列表
文章目录