eval Segmentation fault
yja1 opened this issue · comments
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'