mmpretrain/tests/test_models
Philipp Allgeuer 7dca27dd57
[Fix] Fix warning with `torch.meshgrid`. (#860)
* Fix warning with torch.meshgrid

* Add torch_meshgrid_ij wrapper

* Use `digit_version` instead of packaging package.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-30 15:01:36 +08:00
..
test_backbones [Enhance] RepVGG for YOLOX-PAI. (#1025) 2022-09-30 14:20:53 +08:00
test_utils [Fix] Fix warning with `torch.meshgrid`. (#860) 2022-09-30 15:01:36 +08:00
test_classifiers.py [Improve] Use `forward_dummy` to calculate FLOPS. (#953) 2022-08-08 18:34:09 +08:00
test_heads.py [Feature] Support EfficientFormer. (#954) 2022-08-16 23:38:08 +08:00
test_neck.py [Feature] Support gem pooling (#677) 2022-02-17 02:17:36 +08:00