mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix whl infer ()

pull/696/head
littletomatodonkey 2021-04-21 16:02:09 +08:00 committed by GitHub
parent fec5b5147e
commit 9d74393eb1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

4
paddleclas.py
View File

@ -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'