Thanks for your response. I’ve modified the code slightly to work, and the problem I now face is that the data loader reading from these data sets is way too slow. I’ve elaborated a little more here: GAN Training takes too long!
Please advise.
Regards