mmpretrain/mmcls/models/losses/__init__.py

12 lines
497 B
Python
Raw Normal View History

from .accuracy import Accuracy, accuracy
from .cross_entropy_loss import CrossEntropyLoss, cross_entropy
from .eval_metrics import f1_score, precision, recall
from .label_smooth_loss import LabelSmoothLoss, label_smooth
2020-07-08 21:45:44 +08:00
from .utils import reduce_loss, weight_reduce_loss, weighted_loss
__all__ = [
'accuracy', 'Accuracy', 'cross_entropy', 'CrossEntropyLoss', 'reduce_loss',
'weight_reduce_loss', 'label_smooth', 'LabelSmoothLoss', 'weighted_loss',
'precision', 'recall', 'f1_score'
]