Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

ModelSpeedupv2 bug lists

Jump to bottom
Super Daniel edited this page Mar 21, 2023 · 2 revisions

I tested nni on several mainstream CNN model zoos. The result seems frustrating.

+--------------------------+--------------+----------------------------------+ | package | model name | skip_reason | |--------------------------+--------------+----------------------------------| | timm | beit | cannot trace (torch._assert) | | timm | beitv2 | cannot trace (torch._assert) | | timm | cait | cannot trace (torch._assert) | | timm | coat | cannot trace (torch._assert) | | timm | convit | cannot trace (torch._assert) | | timm | deit | cannot trace (torch._assert) | | timm | dm_nfnet | cannot trace (bool proxy) | | timm | eca_nfnet | cannot prune (empty mask) | | timm | ese_vovnet | cannot trace (diff # of nodes) | | timm | gmixer | cannot trace (torch._assert) | | timm | gmlp | cannot trace (torch._assert) | | timm | hardcorenas | cannot trace (torch._assert) | | timm | mixer | cannot trace (torch._assert) | | timm | nf_ecaresnet | cannot prune (empty mask) | | timm | nf_regnet | cannot trace (torch._assert) | | timm | regnet | cannot trace (torch._assert) | | timm | skresnet | cannot trace (diff # of nodes) | | timm | swin | cannot trace (torch._assert) | | timm | vit | cannot trace (torch._assert) | | torchvision | convnext | cannot prune (empty layer) | | torchvision | googlenet | output is dict | | torchvision | inception | cannot trace | | torchvision | shufflenet | cannot prune (endless loop) | | torchvision | vit | cannot prune (jit trace error) | | torchvision.detection | fasterrcnn | cannot trace (bool proxy) | | torchvision.detection | fcos | cannot trace (bool proxy) | | torchvision.detection | maskrcnn | cannot trace (bool proxy) | | torchvision.detection | retinanet | cannot trace (bool proxy) | | torchvision.detection | ssd | cannot trace (bool proxy) | | torchvision.detection | keypointrcnn | cannot trace (bool proxy) | | torchvision.segmentation | lraspp | cannot prune (fix mask conflict) | | torchvision.video | mc3_18 | cannot prune (empty mask) | | torchvision.video | mvit_v2_s | cannot trace (bool proxy) | | torchvision.video | r2plus1d_18 | cannot prune (empty mask) | | torchvision.video | r3d_18 | cannot prune (empty mask) | | torchvision.video | s3d | cannot prune (empty mask) | +--------------------------+--------------+----------------------------------+

This wiki is a journal that tracks the development of NNI. It's not guaranteed to be up-to-date. Read NNI documentation for latest information: https://nni.readthedocs.io/en/latest/

Clone this wiki locally