美文网首页
load pytorch自带resnet模型并导入自己的pret

load pytorch自带resnet模型并导入自己的pret

作者: 幽并游侠儿_1425 | 来源:发表于2019-07-03 01:17 被阅读0次

1.

导入自带的resnet模型

代码如下:

import torch

import os

import torchvision.models as models

resnet50 = models.resnet50(pretrained = True)

遇到问题:

在Spyder中运行会报错如下:

ValueError: module functions cannot set METH_CLASS or METH_STATIC

但是在命令行中运行就不会报错。

解决办法:

有人遇到过跟我一样的问题:

https://www.gitmemory.com/issue/pytorch/pytorch/2731/466061749

相关文章

网友评论

      本文标题:load pytorch自带resnet模型并导入自己的pret

      本文链接:https://www.haomeiwen.com/subject/awducctx.html