Solved by Aliktk in post #2 
              
              
                Answer: 
After a few try-and-hit methods I find a solution that is very simple to map the weights of the state_dic. 
Which is in the form of strict=False. 
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")


# device = torch.device("cpu")
model = rps.models.get_classifier(model…