I have built a simple CNN using tensorflow-gpu 2.4.0 for cifar10
# Set constent num_classes = len(label_names) # The number of classes input_shape = (32, 32,