Ma Zerun 65410b05ad

Fix Mobilenetv3 structure and add pretrained model (#291) ...

* Refactor Mobilenetv3 structure and add ConvClsHead.

* Change model's name from 'MobileNetv3' to 'MobileNetV3'

* Modify configs for MobileNetV3 on CIFAR10.

And add MobileNetV3 configs for imagenet

* Fix activate setting bugs in MobileNetV3.

And remove bias in SELayer.

* Modify unittest

* Remove useless config and file.

* Fix mobilenetv3-large arch setting

* Add dropout option in ConvClsHead

* Fix MobilenetV3 structure according to torchvision version.

1. Remove with_expand_conv option in InvertedResidual, it should be decided by channels.

2. Revert activation function, should before SE layer.

* Format code.

* Rename MobilenetV3 arch "big" to "large".

* Add mobilenetv3_small torchvision training recipe

* Modify default `out_indices` of MobilenetV3, now it will change
according to `arch` if not specified.

* Add MobilenetV3 large config.

* Add mobilenetv3 README

* Modify InvertedResidual unit test.

* Refactor ConvClsHead to StackedLinearClsHead, and add unit tests.

* Add unit test for `simple_test` of `StackedLinearClsHead`.

* Fix typo

Co-authored-by: Yidi Shao <ydshao@smail.nju.edu.cn>

2021-06-27 23:19:36 +08:00

..

__init__.py

Fix Mobilenetv3 structure and add pretrained model (#291)

2021-06-27 23:19:36 +08:00

base_head.py

[Feature]Delete comments (#298)

2021-06-12 21:45:22 +08:00

cls_head.py

[Fix] Build compatible with low pytorch versions (#301)

2021-06-14 23:25:35 +08:00

linear_head.py

[Bug]Fix linearclshead (#307)

2021-06-16 00:37:16 +08:00

multi_label_head.py

[Fix] Build compatible with low pytorch versions (#301)

2021-06-14 23:25:35 +08:00

multi_label_linear_head.py

[Fix] Build compatible with low pytorch versions (#301)

2021-06-14 23:25:35 +08:00

stacked_head.py

Fix Mobilenetv3 structure and add pretrained model (#291)

2021-06-27 23:19:36 +08:00

vision_transformer_head.py

[Fix] Build compatible with low pytorch versions (#301)

2021-06-14 23:25:35 +08:00