criterion paratemter in the command line of training a new model about top 50 codes

[deweihu96]

Just mention that while training a new model among the top 50 labels, we need to set the criterion as precision_at_5 instead of precision_at_8. Otherwise, in training.py it will not save model.pth file.