Could you update PyTorch to the latest stable release (1.7.1) or the nightly binary and rerun your code?
You might be running into an already fixed issue. If you are still seeing this error, could you post an executable code snippet to reproduce it?