pth模型保存时是按照“整个模型保存”和“只保存模型参数”会影响模型的加载和访问方式

保存方式为“整个模型”(torch.save(model, PATH)):

import torch
if __name__ == '__main__':
    model_pth = r'D:\${modelPath}\${modelName}.pth'
    net = torch.load(model_pth, map_location=torch.device('cpu'))
    for key, value in net["state_dict"].items():
        print(key,value.size(),sep="  ")

输出(部分截图)为:

保存方式为“只保存模型参数”(torch.save(model.state_dict(), PATH)):

待补充

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐