fix ssd quantization script error

example/quantization/README.md

@@ -218,7 +218,7 @@ data/
 |---val.lxt
 |---val.idx
 model/
-|---ssd_vgg16_reduced_300.params
+|---ssd_vgg16_reduced_300-0000.params
 |---ssd_vgg16_reduced_300-symbol.json
 ```
 
@@ -322,4 +322,4 @@ by invoking `launch_quantize.sh`.
 
 **NOTE**: 
 - This example has only been tested on Linux systems.
-- Performance is expected to decrease with GPU, however the memory footprint of a quantized model is smaller. The purpose of the quantization implementation is to minimize accuracy loss when converting FP32 models to INT8. MXNet community is working on improving the performance. 
+- Performance is expected to decrease with GPU, however the memory footprint of a quantized model is smaller. The purpose of the quantization implementation is to minimize accuracy loss when converting FP32 models to INT8. MXNet community is working on improving the performance.

example/ssd/quantization.py

@@ -119,12 +119,14 @@ def save_params(fname, arg_params, aux_params, logger=None):
     exclude_first_conv = args.exclude_first_conv
     excluded_sym_names = []
     rgb_mean = '123,117,104'
-    calib_layer = lambda name: name.endswith('_output')
     for i in range(1,19):
         excluded_sym_names += ['flatten'+str(i)]
     excluded_sym_names += ['relu4_3_cls_pred_conv',
                             'relu7_cls_pred_conv',
-                            'relu4_3_loc_pred_conv']
+                            'relu4_3_loc_pred_conv',
+                            'multibox_loc_pred',
+                            'concat0',
+                            'concat1']
     if exclude_first_conv:
         excluded_sym_names += ['conv1_1']
 
@@ -156,9 +158,9 @@ def save_params(fname, arg_params, aux_params, logger=None):
                                                         ctx=ctx, excluded_sym_names=excluded_sym_names,
                                                         calib_mode=calib_mode, calib_data=eval_iter,
                                                         num_calib_examples=num_calib_batches * batch_size,
-                                                        calib_layer=calib_layer, quantized_dtype=args.quantized_dtype,
+                                                        calib_layer=None, quantized_dtype=args.quantized_dtype,
                                                         label_names=(label_name,),
-                                                        calib_quantize_op = True,
+                                                        calib_quantize_op=True,
                                                         logger=logger)
         sym_name = '%s-symbol.json' % ('./model/cqssd_vgg16_reduced_300')
         param_name = '%s-%04d.params' % ('./model/cqssd_vgg16_reduced_300', epoch)