模型
billionfish opened this issue · 3 comments
billionfish commented
请问预训有大模型吗 resnet那种
github-luffy commented
没用过resnet大模型 不过尝试了resnest50的模型 效果还不错
billionfish commented
试了下预训练的模型
RuntimeError: ./pretrained_model/mobileNetV2_1.0.pth is a zip archive (did you mean to use torch.jit.load()?)
好像模型载入不进去,请问你的torch是什么版本?
github-luffy commented
torch 1.5.0 torchvision 0.6.0