Automatic calculate the number of classes when training

[zacario-li]

number of classes, here the net use 1000 as default, this line code, I suggest that, we can use len(train_loader.dataset.classes).