From 9d74393eb1827d0bc6a09a56df9d5ea92bc60b8f Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Wed, 21 Apr 2021 16:02:09 +0800
Subject: [PATCH] fix whl infer (#692)

---
 paddleclas.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddleclas.py b/paddleclas.py
index d8c4c387b..c4314ec6d 100644
--- 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'