PyTorch Models

Overview

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

python 复制代码
torch.save(vgg16, "vgg16.pt") 
torch.save(vgg16,"vgg16.ckpt") 
torch.save(vgg16,"vgg16.pth") 
torch.save(vgg16,"vgg16.pkl")

Save & Load Models

保存整个模型

python 复制代码
torch.save(model, PATH)
python 复制代码
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="  ")

只保存模型参数

python 复制代码
torch.save(net.state_dict(),path2)
python 复制代码
model.load_state_dict(torch.load(path2))
相关推荐
甜辣uu4 分钟前
【源码讲解+复现】YOLOv10: Real-Time End-to-End Object Detection
人工智能·yolo·目标检测·nms-free
胖达不服输23 分钟前
「日拱一码」105 机器学习原子间势能MLIP
人工智能·机器学习·机器学习原子间势能·mlip
CodeCraft Studio1 小时前
国产化Excel开发组件Spire.XLS教程:将Python列表转换为Excel表格(3种实用场景)
开发语言·python·excel·spire.xls·python列表转excel·国产化文档开发
企鹅侠客1 小时前
基于python写的PDF表格提取到excel文档
python·pdf·excel·pdf文档表格转excel
mortimer1 小时前
Python 中那些鲜为人知但实用的工具函数
python
weixin_421133411 小时前
Django 的文档接口
python·django·sqlite
sight-ai1 小时前
超越基础:SightAI 智能路由与多模型选择实战
人工智能·开源·大模型·api
OpenVINO 中文社区1 小时前
端侧AI创新挑战赛正式启动
人工智能
取酒鱼食--【余九】1 小时前
深度学习经典网络解析:ResNet
网络·人工智能·深度学习·神经网络·resnet·卷积神经网络·残差神经网络
搞科研的小刘选手1 小时前
【通信&网络安全主题】第六届计算机通信与网络安全国际学术会议(CCNS 2025)
大数据·人工智能·网络安全·vr·通信工程·网络技术·计算机工程