mmclassification/resnet50_b16x8_cifar10_mixup.py at c5374854aa13c0e4fa08526f3499bcb5eb2a5b4a - mmclassification - Gitea: Git with a cup of tea

mirrors/mmclassification

mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 16:24:29 +08:00

ftbabi bdd6b01ae7

[Feature] Add "mixup" from Bag of Tricks (#160 )

* 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

2021-02-25 14:06:58 +08:00

6 lines

178 B

Python

Raw Blame History

 _base_ = [
     '../_base_/models/resnet50_cifar_mixup.py',
     '../_base_/datasets/cifar10_bs16.py',
     '../_base_/schedules/cifar10_bs128.py', '../_base_/default_runtime.py'
 ]