9f7168e770
Add RegNetX configs ( #269 )
...
* Add RegNetX configs
* Fix typo in RegNet readme.
* Use comment to mark tag.
2021-06-04 14:29:13 +08:00
759aa92437
[Bug]add cutmix_prob to config ( #220 )
...
* add cutmix_prob to config
* empty
2021-04-21 21:57:09 +08:00
31a6a362ba
Add some vit configs ( #217 )
...
* add vit_base_patch32_384_finetune.py
* add vit_base_patch32_384_finetune_imagenet.py to vision_transformer
* add vit_large_patch16_384_finetune.py to models
* add vit_large_patch16_384_finetune_imagenet.py to vision_transformer
* add vit_large_patch32_384_finetune to models
* add vit_large_patch32_384_finetune_imagenet to vision_transformer
* add vit_large_patch16_224_finetune.py to models
* add vit_large_patch16_224_finetune_imagenet.py to vision_transformer
* delete some useless comments
2021-04-20 11:32:20 +08:00
affb39fe07
[Feature]Add Vit ( #214 )
...
* add imagenet bs 4096
* add vit_base_patch16_224_finetune
* add vit_base_patch16_224_pretrain
* add vit_base_patch16_384_finetune
* add vit_base_patch16_384_finetune
* add vit_b_p16_224_finetune_imagenet
* add vit_b_p16_224_pretrain_imagenet
* add vit_b_p16_384_finetune_imagenet
* add vit
* add vit
* add vit head
* vit unitest
* keep up with ClsHead
* test vit
* add flag to determiine whether to calculate acc during training
* Changes related to mmcv1.3.0
* change checkpoint saving interval to 10
* add label smooth
* default_runtime.py recovery
* docformatter
* docformatter
* delete 2 lines of comments
* delete configs/_base_/schedules/imagenet_bs4096.py
* add configs/_base_/schedules/imagenet_bs2048_AdamW.py
* rename imagenet_bs4096.py to imagenet_bs2048_AdamW.py
* add helpers.py
* test vit hybrid backbone
* fix HybridEmbed
* use to_2tuple instead
2021-04-16 19:22:41 +08:00
1cde6f6e65
[Feature] Add cutmix option ( #198 )
...
* Add cutmix option
* fix code style
* add some annotations
* add annotation about custom_hooks
* check constraint of alpha > 0
* add test cutmix
* fix code style
* add cutmix to configs/models
* add cutmix to configs/resnet
* flake8
* empty
2021-04-14 21:27:42 +08:00
af83e981ac
[Bug]Fix label smooth bug ( #203 )
...
* 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
2021-04-13 13:53:56 +08:00
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
bc1b08ba41
Add VGG and pretained models ( #27 )
...
* add vgg
* add vgg model coversion tool
* fix out_indices and docstr
* add vgg models in configs
* add params, flops and accuracy in docs
* add pretrained models url
* use ConvModule and refine var names
* update vgg conversion tool
* modify bn config
* add docs for arch_setting
* add unit test for vgg
* rm debug code
* update vgg pretrained models
2020-09-29 17:49:42 +08:00
8d3acce307
Add ResNeSt ( #25 )
...
* Add ResNeSt
* fixed test
* refactor
* add ResNeSt base
* update modelzoo
* update modelzoo
* Add S-200,S269 _base_
2020-09-22 10:41:51 +08:00
4b46fd6dc7
Add CIFAR10 configs and models ( #38 )
...
* rename cifar configs
* add cifar10 configs
* fix linting
* add params, flops and accuracy in docs
* add cifar10 pretrained models and logs
* use oss-accelerate url
* del unused cifar10 configs
2020-08-26 16:34:15 +08:00
18ac0f0fa3
Update shufflenet config ( #31 )
...
* update shufflenet config
* remove old config
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 10:46:39 +08:00
5c88675a71
Add configs for MNIST, CIFAR10 and ImageNet
2020-07-12 00:06:57 +08:00
Ma Zerun
whcao
ftbabi
Lei Yang
Jerry Jiarui XU
Xiaojie Li
yanglei