Could you post all arguments to create the model as well as all necessary input shapes to execute it?
Also, could you post your current setup, i.e.

GPU
CUDA, cudnn version
PyTorch version and you’ve installed it