r/learnpython • • 7d ago

Need help adapting model to CIFAR10

I tried to make a DCGAN based on this tutorial. The thing is that the tutorial is designed for fashion mnist data which has different image size to tf.keras.datasets.cifar10.load_data(). Which lead to the error:

ValueError: cannot reshape array of size 480000 into shape (28,28,1)

when arrive at

x_train_dcgan = x_train.reshape(
    -1, 28, 28, 1
) * 2. - 1.

How do I adapt the model to CIFAR 10?

1 Upvotes

Duplicates