d-li14 / mobilenetv3.pytorch

74.3% MobileNetV3-Large and 67.2% MobileNetV3-Small model on ImageNet

Home Page:https://arxiv.org/abs/1905.02244

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

eval Segmentation fault

yja1 opened this issue · comments

commented

import torch
from mobilenetv3 import mobilenetv3_small

net_small = mobilenetv3_small()
net_small.load_state_dict(torch.load('mobilenetv3-small-c7eb32fe.pth'))
net_small.eval()
x = torch.randn(2,3,224,224)
net_small(x)

!!!Segmentation fault (core dumped) in line net_small(x)
torch.version '0.4.1'