basicTrainer.py 文件源码

def loadInfo(self, folder, model, state, smanager):
        if(not os.path.exists(folder)):
            return (model, state, 1)
        list_files = []
        model_name = model.getName()
        for file in os.listdir(folder):
            if(file.startswith(model_name) and file.endswith(".state")):
                list_files.append(file)
        if(len(list_files) > 0):
            sorted_list = self.natural_sort(list_files)
            fname_state = sorted_list[-1]

            bname = re.split('\.',fname_state)[0]
            fname_model = bname + '.model'
            fname_stats = bname + '.stats'
            epoch = int(re.split('_|\.', bname)[-1]) + 1
            serializers.load_npz(folder + '/' + fname_state, state)
            serializers.load_npz(folder + '/' + fname_model, model)
            smanager.load(folder + '/' + fname_stats)

        else:
            epoch = 1
            # no prev. models...
        return (model, state, epoch)