Ma Zerun 1a7cebe4b9 [Refactor] Refactor unittest (#321) ... * Refactor unit tests folder structure. * Remove label smooth and Vit test in `test_classifiers.py` * Rename test_utils in dataset to test_dataset_utils * Split test_models/test_utils/test_utils.py to multiple sub files. * Add unit tests of classifiers and heads * Use patch context manager. * Add unit test of `is_tracing`, and add warning in `is_tracing` if torch verison is smaller than 1.6.0		2021-07-08 22:49:05 +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	[Refactor] Refactor unittest (#321)	2021-07-08 22:49:05 +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