mmrazor/tests/test_models
LKJacky 122ee38d69
fix bug (#407)
* add resnet50

* fix bug

* fix bug

* fix bug

* refine

* fix bug

* add choice and mask of units to checkpoint (#397)

* add choice and mask of units to checkpoint

* update

* fix bug

* remove device operation

* fix bug

* fix circle ci error

* fix error in numpy for circle ci

* fix bug in requirements

* restore

* add a note

* a new solution

* save mutable_channel.mask as float for dist training

* refine

* mv meta file test

Co-authored-by: liukai <your_email@abc.example>
Co-authored-by: jacky <jacky@xx.com>

* fix bug

* add assert

* fix bug

* change iter to epoch

* bn_imp use abs

Co-authored-by: jacky <jacky@xx.com>
Co-authored-by: liukai <your_email@abc.example>
2022-12-22 17:41:10 +08:00
..
test_algorithms fix bug (#407) 2022-12-22 17:41:10 +08:00
test_architectures [Improvement] Adapt OFA series with SearchableMobileNetV3 (#385) 2022-12-15 22:19:55 +08:00
test_classifier [Feature] Add Autoformer algorithm (#315) 2022-11-14 13:01:04 +08:00
test_distillers
test_losses [Feature] PyTorch version of `PKD: General Distillation Framework for Object Detectors via Pearson Correlation Coefficient`. (#304) 2022-10-26 19:18:20 +08:00
test_mutables Enhance the Abilities of the Tracer for Pruning. (#371) 2022-12-08 15:59:27 +08:00
test_mutators [Improvement] Adapt OFA series with SearchableMobileNetV3 (#385) 2022-12-15 22:19:55 +08:00
test_subnet [Improvement] Adapt OFA series with SearchableMobileNetV3 (#385) 2022-12-15 22:19:55 +08:00
test_task_modules [Improvement] Adapt OFA series with SearchableMobileNetV3 (#385) 2022-12-15 22:19:55 +08:00
__init__.py Enhance the Abilities of the Tracer for Pruning. (#371) 2022-12-08 15:59:27 +08:00