Automatic calculate the number of classes when training #15
Description
number of classes, here the net use 1000 as default,
this line code, I suggest that, we can use len(train_loader.dataset.classes).
{{ message }}
number of classes, here the net use 1000 as default,
this line code, I suggest that, we can use len(train_loader.dataset.classes).