Skip to content

Commit

Permalink
remove unsupported models
Browse files Browse the repository at this point in the history
  • Loading branch information
Laurawly committed Mar 20, 2019
1 parent d6bfd93 commit a7b5d9e
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions tutorials/frontend/deploy_ssd_gluoncv.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,16 +42,14 @@
# x86 conv2d schedule doesn't support dilation.

supported_model = [
'ssd_512_resnet18_v1_voc',
'ssd_512_resnet18_v1_coco',
'ssd_512_resnet50_v1_voc',
'ssd_512_resnet50_v1_coco',
'ssd_512_resnet101_v2_voc',
'ssd_512_mobilenet1.0_voc',
'ssd_512_mobilenet1.0_coco',
]

model_name = supported_model[4]
model_name = supported_model[0]
dshape = (1, 3, 512, 512)
target_list = ctx_list()

Expand Down

0 comments on commit a7b5d9e

Please sign in to comment.