|
|
@@ -17,8 +17,71 @@ import os
|
|
|
from ....utils import logging
|
|
|
|
|
|
CINN_WHITELIST = [
|
|
|
- # TODO: update models based testing result
|
|
|
+ "CLIP_vit_base_patch16_224",
|
|
|
+ "MobileNetV2_x1_0",
|
|
|
+ "PP-HGNet_small",
|
|
|
+ "PP-LCNet_x1_0",
|
|
|
+ "ResNet50",
|
|
|
+ "SwinTransformer_base_patch4_window7_224",
|
|
|
+ "Mask-RT-DETR-L",
|
|
|
+ "RT-DETR-L",
|
|
|
+ "Deeplabv3-R50",
|
|
|
+ "Deeplabv3_Plus-R50",
|
|
|
+ "PP-LiteSeg-T",
|
|
|
+ "SLANet",
|
|
|
+ "PP-OCRv4_mobile_rec",
|
|
|
+ "PP-OCRv4_server_rec",
|
|
|
+ "Nonstationary_ad",
|
|
|
+ "PatchTST_ad",
|
|
|
+ "TimesNet_ad",
|
|
|
+ "Nonstationary",
|
|
|
+ "PatchTST",
|
|
|
+ "RLinear",
|
|
|
+ "TiDE",
|
|
|
+ "TimesNet",
|
|
|
+ "CLIP_vit_large_patch14_224",
|
|
|
+ "MobileNetV2_x0_25",
|
|
|
+ "MobileNetV2_x0_5",
|
|
|
+ "MobileNetV2_x1_5",
|
|
|
+ "MobileNetV2_x2_0",
|
|
|
+ "PP-HGNet_tiny",
|
|
|
+ "PP-HGNet_base",
|
|
|
"PP-LCNet_x0_25",
|
|
|
+ "PP-LCNet_x0_35",
|
|
|
+ "PP-LCNet_x0_5",
|
|
|
+ "PP-LCNet_x0_75",
|
|
|
+ "PP-LCNet_x1_5",
|
|
|
+ "PP-LCNet_x2_5",
|
|
|
+ "PP-LCNet_x2_0",
|
|
|
+ "ResNet18",
|
|
|
+ "ResNet34",
|
|
|
+ "ResNet101",
|
|
|
+ "ResNet152",
|
|
|
+ "ResNet18_vd",
|
|
|
+ "ResNet34_vd",
|
|
|
+ "ResNet50_vd",
|
|
|
+ "ResNet101_vd",
|
|
|
+ "ResNet152_vd",
|
|
|
+ "ResNet200_vd",
|
|
|
+ "SwinTransformer_tiny_patch4_window7_224",
|
|
|
+ "SwinTransformer_small_patch4_window7_224",
|
|
|
+ "SwinTransformer_base_patch4_window12_384",
|
|
|
+ "SwinTransformer_large_patch4_window7_224",
|
|
|
+ "SwinTransformer_large_patch4_window12_384",
|
|
|
+ "Mask-RT-DETR-H",
|
|
|
+ "Mask-RT-DETR-X",
|
|
|
+ "Mask-RT-DETR-M",
|
|
|
+ "Mask-RT-DETR-S",
|
|
|
+ "RT-DETR-H",
|
|
|
+ "RT-DETR-X",
|
|
|
+ "RT-DETR-R18",
|
|
|
+ "RT-DETR-R50",
|
|
|
+ "Deeplabv3-R101",
|
|
|
+ "Deeplabv3_Plus-R101",
|
|
|
+ "PP-LiteSeg-B",
|
|
|
+ "SLANet_plus",
|
|
|
+ "PP-OCRv4_server_rec",
|
|
|
+ "PP-OCRv4_server_rec_doc",
|
|
|
]
|
|
|
|
|
|
|
gaotingquan