pretrained-models.pytorch
pretrained-models.pytorch copied to clipboard
Adaptive pooling for senet
Needed to add this line to make it work on other resolution images.
Oops...Looks like I broke something. I'll take a look at it when I get chance...
@erikgaas For some reason Adaptive pooling doesn't seem to work on all pytorch versions. You can check my workaround which will work for even older versions for this issue in #122