mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
bdd6b01ae7
* Add mixup option * Modify the structure of mixup and add configs * Clean configs * Add test for mixup and SoftCrossEntropyLoss * Add simple test for ImageClassifier * Fix bug in test_losses.py * Add assertion in CrossEntropyLoss
11 lines
310 B
Python
11 lines
310 B
Python
from .channel_shuffle import channel_shuffle
|
|
from .inverted_residual import InvertedResidual
|
|
from .make_divisible import make_divisible
|
|
from .mixup import BatchMixupLayer
|
|
from .se_layer import SELayer
|
|
|
|
__all__ = [
|
|
'channel_shuffle', 'make_divisible', 'InvertedResidual', 'BatchMixupLayer',
|
|
'SELayer'
|
|
]
|