diff --git a/tutorials/frontend/deploy_ssd_gluoncv.py b/tutorials/frontend/deploy_ssd_gluoncv.py index 7dae610471eb..e0d678d9c0f6 100644 --- a/tutorials/frontend/deploy_ssd_gluoncv.py +++ b/tutorials/frontend/deploy_ssd_gluoncv.py @@ -42,8 +42,6 @@ # 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', @@ -51,7 +49,7 @@ 'ssd_512_mobilenet1.0_coco', ] -model_name = supported_model[4] +model_name = supported_model[0] dshape = (1, 3, 512, 512) target_list = ctx_list()