Refer to thread Convert pytorch model to ptl
You might have to check if your model definition has variable number of arguments
Refer to thread Convert pytorch model to ptl
You might have to check if your model definition has variable number of arguments