basicTrainer.py 文件源码

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)