Error training images for AI art generation (create a .ckpt file)

2

You are probably slicing a tensor and modiying it overiding the data directly.
It’s diff to help u without the source code.

Best
J