@ptrblck I will start on that. I have another question though. Digging around it seems that the issue was not present in PyTorch 1.2 . I thought of downgrading to PyTorch 1.2 but as soon as I do that, I would get an error of
image1469×183 97.7 KB
For one of the PyBind modules. This error was not there when I worked with PyTorch 1.3 and above. Any ideas about this?