Hello,
I have saved my Pytorch model as a state_dict, and am now trying to load it into flask for inference.
However when I run my flask API, it just stopped. And using my debugger got me here.
Screenshot from 2021-04-27 15-53-22727×230 22.8 KB

The same model loads in my notebook. Why is this happening? Any help would be great!