Improve clarity to how number of classes is automatically assigned

[chawins]

Currently, args.num_classes does not need to be set because it is automatically handled in each dataloader. This has not been made explicit.