Image classifcation mkldnn

haohuanw · Mar 19, 2019 · 30bfab2 · 30bfab2
1 parent 5016170
commit 30bfab2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/example/quantization/imagenet_gen_qsym_mkldnn.py b/example/quantization/imagenet_gen_qsym_mkldnn.py
@@ -179,6 +179,7 @@ def save_params(fname, arg_params, aux_params, logger=None):
         prefix, epoch = download_model(model_name=args.model, logger=logger)
         sym, arg_params, aux_params = mx.model.load_checkpoint(prefix, epoch)
 
+    sym.save("model_pre_quantize.json")
     sym = sym.get_backend_symbol('MKLDNN')
     sym = sym.get_backend_symbol('MKLDNN_FC')