I don’t think downgrading is an optimal solution. I would rather want to fix the issue using the latest release.
Which PyTorch version are you currently using? If < 0.4.1, could you update to it?
Also, could you share a bit of executable code so that we can have a look at this problem?