I don’t get an error, the loss just never changes and it saves the same picture over and over again.
I’ve decided to rewrite the code without the custom losses inserted into the network and just calculate them in place (this is a bit more PyTorch-y?). I’ll report back in a few days…