fix mxnet rn50 example bug
@@ -290,8 +290,8 @@ if __name__ == '__main__':
calib_data = mx.io.ImageRecordIter(path_imgrec=dataset,label_width=1,preprocess_threads=data_nthreads,batch_size=batch_size,data_shape=data_shape,label_name=label_name,rand_crop=False,rand_mirror=False,shuffle=args.shuffle_dataset,shuffle_chunk_seed=args.shuffle_chunk_seed,seed=args.shuffle_seed,dtype=data_layer_type,ctx=args.ctx,**combine_mean_std)
cnn_tuner = Tuner("./rn50.yaml")
cnn_tuner.tune(fp32_model, q_dataloader=calib_data, eval_dataloader=data)
- return
+ '''
if args.low_precision:
sym, arg_params, aux_params = low_precison_convert(symbol_file,
args.low_precision,
@@ -305,6 +305,7 @@ if __name__ == '__main__':
logger.info('Running model %s for inference' % symbol_file)
score(sym, arg_params, aux_params, data, [ctx], label_name,
max_num_examples=num_inference_images, logger=logger)
+ '''
else:
logger.info('Running model %s for inference' % symbol_file)
speed = benchmark_score(symbol_file, ctx, batch_size,
