simclr algorithm implementation gets error: “one of the variables needed for gradient computation has been modified by an inplace operation”

Question

I\'m trying to follow a guide and implement ths simclr algorithm but I keep getting this message:

RuntimeError: one of the variables needed for gradient computation has b