davidADSP / Generative_Deep_Learning_2nd_Edition

The official code repository for the second edition of the O'Reilly book Generative Deep Learning: Teaching Machines to Paint, Write, Compose and Play.
https://www.oreilly.com/library/view/generative-deep-learning/9781098134174/
Apache License 2.0
990 stars 363 forks source link

Applying CGANs on more than 2 classes #27

Open Zardian18 opened 6 months ago

Zardian18 commented 6 months ago

Following the book, ive been trying to apply ConditionalWGAN for fashion mnist dataset with 10 classes but am facing error. I changed the model accordingly for both generator and critic and instead of one hot encoding the labels, im passing them just as numbers. What are the changes I am supposed to make. I've tried a few but none works. And if i try to fit it normally, this error occurs. Does anyone know how to implement it? image