adomakor412:

How did you figure out my initial model would work for an input shape of [batch_size, 3, 32, 32]?


It was a guess based on your model definition.
The indexing error is raised in plot_classes_preds, where classes[preds[idx]] uses an invalid index.
Could you check the shape of preds and the value of idx as well as the shape of classes and the value of preds[idx] as one of them is using an invalid indexing?