Initialize your model in the same way (with pretrained set to True or False) and check if the model is working at all. Your initial question seemed to target the model loading specifically, but I’m unsure if the setup you are using on the “Nvidia-ProAI” device is working at all with this model or not.