mirror of
https://github.com/open-mmlab/mmpretrain.git
synced 2025-06-03 06:40:40 +08:00
af83e981ac
* add convert_to_one_hot * add test_label_smooth_loss * add my label_smooth_loss * fix CELoss bug * test new label smooth loss * LabelSmoothLoss downward compatibility * add some comments * remove the old version of LabelSmoothLoss * add some comments * add some comments * add some comments * add label smooth to config
16 lines
714 B
Python
16 lines
714 B
Python
from .accuracy import Accuracy, accuracy
|
|
from .asymmetric_loss import AsymmetricLoss, asymmetric_loss
|
|
from .cross_entropy_loss import (CrossEntropyLoss, binary_cross_entropy,
|
|
cross_entropy)
|
|
from .focal_loss import FocalLoss, sigmoid_focal_loss
|
|
from .label_smooth_loss import LabelSmoothLoss
|
|
from .utils import (convert_to_one_hot, reduce_loss, weight_reduce_loss,
|
|
weighted_loss)
|
|
|
|
__all__ = [
|
|
'accuracy', 'Accuracy', 'asymmetric_loss', 'AsymmetricLoss',
|
|
'cross_entropy', 'binary_cross_entropy', 'CrossEntropyLoss', 'reduce_loss',
|
|
'weight_reduce_loss', 'LabelSmoothLoss', 'weighted_loss', 'FocalLoss',
|
|
'sigmoid_focal_loss', 'convert_to_one_hot'
|
|
]
|