prlz77 / ResNeXt.pytorch

Reproduces ResNet-V3 with pytorch
MIT License
505 stars 121 forks source link

cannot run inference mismatch sizes #17

Open kirk86 opened 5 years ago

kirk86 commented 5 years ago

Hi, I was trying to run inference on the trained model using the test.py script but first there's an error on ordered dict iteritems() method which should be chnaged into items() and the other thing is that there are a lot of mismatches when loading the weighst into the model. Here's a screenshot:

image

Any ideas how to resolve those?

prlz77 commented 5 years ago

Hi @kirk86 there was an update on train.py that wasn't reflected in test.py. I have updated it. Thanks for pointing it out :)

https://github.com/prlz77/ResNeXt.pytorch/blob/master/test.py#L73

ma3252788 commented 4 years ago

The same problem!!!