mmpretrain/mmcls/models/losses/__init__.py

10 lines
404 B
Python
Raw Normal View History

from .accuracy import Accuracy, accuracy
from .cross_entropy_loss import CrossEntropyLoss, cross_entropy
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',
2020-07-08 21:45:44 +08:00
'weight_reduce_loss', 'label_smooth', 'LabelSmoothLoss', 'weighted_loss'
]