定義了一個類BaseNet并實例化該類：

net=BaseNet()

保存net時報錯object has no attribute 'state_dict'

torch.save(net.state_dict(), models_dir)

原因是定義類的時候不是繼承nn.Module類，比如：

class BaseNet(object):def __init__(self):

把類定義改為

class BaseNet(nn.Module):def __init__(self):super(BaseNet, self).__init__()

?

總結(jié)

以上是生活随笔為你收集整理的pytorch保存模型时报错***object has no attribute 'state_dict'的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。