mmclassification/mmcls/models/utils
Ma Zerun 71621a5f62
Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 (#347)
2021-07-07 11:55:53 +08:00
..
augment [Feature]Add augments to models/utils (#278) 2021-06-20 09:44:51 +08:00
__init__.py Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 (#347) 2021-07-07 11:55:53 +08:00
attention.py [Feature] Add swin-transformer model. (#271) 2021-07-01 09:30:42 +08:00
channel_shuffle.py Dev/backbone utils 2020-06-15 16:42:15 +08:00
embed.py [Feature] Add swin-transformer model. (#271) 2021-07-01 09:30:42 +08:00
helpers.py Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 (#347) 2021-07-07 11:55:53 +08:00
inverted_residual.py Fix Mobilenetv3 structure and add pretrained model (#291) 2021-06-27 23:19:36 +08:00
make_divisible.py [Fix] Fix optional issues in docstring (#138) 2021-01-14 11:09:08 +08:00
se_layer.py Fix Mobilenetv3 structure and add pretrained model (#291) 2021-06-27 23:19:36 +08:00