Permute and Reshape causing issues with Detach

[zetyquickly (Emil Bogomolov)]

Hi there,

Have a look at this issue

Nightly version of PyTorch doesn’t have this bug. Install it via
!pip3 install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu111/torch_nightly.html -U
in your colab