pytorch加载.pth文件

最新推荐文章于 2025-04-02 13:13:18 发布

柠檬树下你和我₰

最新推荐文章于 2025-04-02 13:13:18 发布

阅读量3.6k

点赞数 4

文章标签： pytorch python

本文链接：https://blog.csdn.net/m0_52347246/article/details/120201452

版权

该博客介绍了如何在Python环境中使用PyTorch库加载.pth文件。通过创建新的python文件并导入torch，然后指定.pth文件的路径，可以加载并打印模型的详细内容。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

pytorch加载.pth文件

1.创建一个新的python文件
2.键入代码，运行。
注意：
profile后面为所要查看的pth文件地址。
代码：
import torch
pthfile = r’D:\deep reinforcemnt learning\PRUNING FILTERS FOR EFFICIENT CONVNETS\Pruning_filters_for_efficient_convnets-master\trained_models\check_point.pth’
net = torch.load(pthfile)
print(net)