Pytorch model problems

[Grigory Samokhin]

Hello! Thanks for your work!
I have some problem when running model.py file

And what means "geffnet.mobilenetv3._gen_mobilenet_v3 needs to be patched to return the parameters instead of instantiating the network" ?

[Emiliana]

Hi! It means that here, model_kwargs needs to be returned instead of model. After that change is made, it should run properly.