MobileNetV2-pytorch
MobileNetV2-pytorch copied to clipboard
Automatic calculate the number of classes when training
number of classes, here the net use 1000 as default,
this line code, I suggest that, we can use len(train_loader.dataset.classes)
.