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

2019-07-03  本文已影响0人  幽并游侠儿_1425

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

上一篇 下一篇

猜你喜欢

热点阅读