search

implus / PytorchInsight

a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results
858 stars 123 forks source link

pretrained model #9

Closed ztyxd closed 5 years ago

ztyxd commented 5 years ago

I have download pretrained model : resnet-sge_101.pth.tar， but I cannot open extract this file in both windows and ubuntu. Can U give me some suggestions about this question or other links of the pretrained models? Thanks!

implus commented 5 years ago

You do not need to extract this file. Please refer to https://github.com/implus/PytorchInsight/blob/master/classification/pretrain_test.md

Simply use ``python -W ignore imagenet.py -a sge_resnet101 --data /yourpath/imagenet1k/ --gpu-id 0 -e --resume ../pretrain/sge_resnet101.pth.tar'' will be ok.

ztyxd commented 5 years ago

Thanks a lot，I solve this problem.