@ArshadIram
Hi, the issue is since the input size of CIFAR dataset is (32 x 32) while that of MNIST is (28 x 28). You need to change the stride or kernel size or padding (you can change any one of them).
Output shape of a conv = (W - F + 2P)/S  + 1. So in your make_gen_block function change the kernel_size or stride. Or add padding parameter.