pytorch模型保存

模型保存

    def save_model(self):
        torch.save(self.state_dict(), self.file_path)

    def load_model(self):
        if os.os.path.exists(self.file_path):
            print("----------loading model----------")
            self.load_state_dict(torch.load(self.file_path))
        else:
            print('没有预训练模型')
已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 游动-白 设计师:白松林 返回首页