Unfortunately, I cannot share the Generator model’s code at this time. I do understand the problem that poses, but I am confident that the generator’s structure is not at fault. That is because the network trains normally without error when I don’t federate it, under the same circumstances.