This repository was archived by the owner on Sep 18, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 1.9k
ModelSpeedupv2 bug lists
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/