Hi there,
I got a weird issue. When using Pytorch 1.4 my code runs just fine.
But when using 1.5.1 I am getting the RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation’.
I am using CrossEntropyLoss.
Cheers,
Tassilo