Can this project deal with datasets with over 2 classes?

[zetn]

Hello, I wanna ask about if this project can dealt with datasets with over 2 classes, Like cifar10?

[duggalrahul]

Yes.
Just set the nb_classes variable to the number of classes you want. I'm not sure if AlexNet can accept inputs as small as in the CIFAR 10 dataset. You can possibly remove some pooling layers to accommodate small sized image.

Best,
Rahul

[G33kyKitty]

Hello, why did you use nb_classes to be 2??

[Gokkulnath]

@G33kyKitty Because he has used Pre-trained weights for Binary Classification. (Cat/Dog)

[G33kyKitty]

Ok thank you @Gokkulnath ..