Not downloading correctly pre-trained models except mn10

[JNaranjo-Alcazar]

Hi,
first of all, congrats for the amazing job here.
I am trying to get predictions of different models but I cannot download directly pretrained models according to the repo:

model = get_mn(pretrained_name="mn20_as")

triggers an error when loading the state dict. I think the code is hard-coded for mn10_as only.

Is that the case? How can I download and load the other pretrained models properly?

Thanks in advance

[paullee137]

I encountered the same problem and fixed it by adding the provided config:

model = get_mn(pretrained_name="mn20_as", width_mult=2.0)