Hi @ptrblck ,
unfortunately, I can not share the entire code (sorry, not my decision).
However, the generator is an encoder/decoder network (u-net like) that accepts in input a tensor of [batch_size, 1, 256, 256] and returns an image of the same shape.
As I said, if I use the CPU it runs smoothly, so the size match should be ok.
If it can help, I tried with a VGG (instead of ResNet) and I get the same error.
Thanks a lot!