Solved by Eric_Hou in post #5 
              
              
                When you have to inference with a pytorch version below 1.6, try code below to convert your model, because pytorch changed the model saving format after version 1.6. 
torch.save(model.state_dict(), path, _use_new_zipfile_serialization=False)