mmpretrain/tests/test_models
Hakjin Lee 05e4bc17b2
[Feature] Support Activation Checkpointing for ConvNeXt. (#1152)
* Support Activation Checkpointing for ConvNeXt

* Add test case

* Lint

* Add docstring
2022-11-14 17:08:31 +08:00
..
test_backbones [Feature] Support Activation Checkpointing for ConvNeXt. (#1152) 2022-11-14 17:08:31 +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