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
..
__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	Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 (#347)	2021-07-07 11:55:53 +08:00
linear_head.py	Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 (#347)	2021-07-07 11:55:53 +08:00
multi_label_head.py	Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 (#347)	2021-07-07 11:55:53 +08:00
multi_label_linear_head.py	Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 (#347)	2021-07-07 11:55:53 +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	Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 (#347)	2021-07-07 11:55:53 +08:00