It crashes

[RyannDaGreat]

When I try to run resnet_cppn.py, it crashes with the following error:

RuntimeError: set_sizes_and_strides is not allowed on a Tensor created from .data or .detach().
If your intent is to change the metadata of a Tensor (such as sizes / strides / storage / storage_offset)
without autograd tracking the change, remove the .data / .detach() call and wrap the change in a `with torch.no_grad():` block.
For example, change:
    x.data.set_(y)
to:
    with torch.no_grad():

It happened on my computer, and also on Google Colab. Here's the link:

https://colab.research.google.com/drive/14VfAbTGJKgtehWG9eV9cSXdI9rZHLjxH?usp=sharing

It crashes before completing a single iteration of training