fix whl infer (#692)

9d74393e · littletomatodonkey · GitHub · fec5b514 · 9d74393e
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

paddleclas.py paddleclas.py +2 -2

未找到文件。
--- a/paddleclas.py
+++ b/paddleclas.py
@@ -88,8 +88,8 @@ model_names = {
    'MobileNetV3_large_x0_75', 'MobileNetV3_small_x0_5', 'DenseNet169',
    'EfficientNetB5', 'DeiT_base_distilled_patch16_224',
    'DeiT_base_distilled_patch16_384', 'DeiT_base_patch16_224',
-    'DeiT_base_patch16_384_infer', 'DeiT_small_distilled_patch16_224',
-    'DeiT_small_patch16_224_infer', 'DeiT_tiny_distilled_patch16_224',
+    'DeiT_base_patch16_384', 'DeiT_small_distilled_patch16_224',
+    'DeiT_small_patch16_224', 'DeiT_tiny_distilled_patch16_224',
    'DeiT_tiny_patch16_224', 'ViT_base_patch16_224', 'ViT_base_patch16_384',
    'ViT_base_patch32_384', 'ViT_large_patch16_224', 'ViT_large_patch16_384',
    'ViT_large_patch32_384', 'ViT_small_patch16_224'