mmclassification/mmcls/models
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
..
backbones [Feature] Add swin-transformer model. (#271) 2021-07-01 09:30:42 +08:00
classifiers fix a bug when samples_per_gpu==1 (#311) 2021-06-30 20:57:21 +08:00
heads Add `is_tracing` helper function to fix a tracing bug in PyTorch 1.6 (#347) 2021-07-07 11:55:53 +08:00
losses
necks
utils 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
builder.py