basicTrainer.py 文件源码

def saveInfo(self, model, optimizer, smanager, epoch, outputFolder, saveEach):
        #ipdb.set_trace()
        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)
            smanager.save(bname + '.stats')