'ScriptModel' object has no attribute 'save'

最新推荐文章于 2024-08-18 18:35:16 发布

AI算法网奇

最新推荐文章于 2024-08-18 18:35:16 发布

阅读量3.8k

点赞数

分类专栏： c++入门宝典 pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/93387032

版权

pytorch知识宝典同时被 2 个专栏收录

499 篇文章 227 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

c++入门宝典

357 篇文章 123 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

在使用PyTorch的torch0.4.0版本时，遇到了'ScriptModel'对象没有'save'属性的错误。这表明在该版本中，无法直接对ScriptModel实例进行保存操作。

摘要由CSDN通过智能技术生成

'ScriptModel' object has no attribute 'save'

原因：torch0.4.0没有保存功能


import torch

class ScriptModel(torch.jit.ScriptModule):
    def __init__(self, N, M):
        super(ScriptModel, self).__init__()
        self.weight = torch.nn.Parameter(torch.rand(N, M))

        self.conv1 = torch.nn.Conv2d(3, 64, kernel_size=7, stride=2, padding=3,
                               bias=False)
        # self.bn1 = torch.nn.BatchNorm2d(64, momentum=0.1)
        self.bn1 = torch.nn.BatchNorm2d(64)
        self.relu = torch.nn.ReLU(inplace=True)
        self.maxpool = torch.nn.MaxPool2d(kernel_size=3, stride=2, padding=1)

        @torch.jit.script_met

了解本专栏