You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by re...@apache.org on 2019/02/07 18:09:21 UTC
[incubator-mxnet] branch master updated: Exclude concat layer for
gpu quantization (#14060)
This is an automated email from the ASF dual-hosted git repository.
reminisce pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new 26ca37c Exclude concat layer for gpu quantization (#14060)
26ca37c is described below
commit 26ca37c41ef93426c43cb28df8691a6c24a04a11
Author: rajeshii <46...@users.noreply.github.com>
AuthorDate: Thu Feb 7 10:08:56 2019 -0800
Exclude concat layer for gpu quantization (#14060)
* exclude concat for gpu quantization
* remove quantized_concat test in non-subgraph flow
---
example/quantization/imagenet_gen_qsym.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/example/quantization/imagenet_gen_qsym.py b/example/quantization/imagenet_gen_qsym.py
index 8a2818c..41713c3 100644
--- a/example/quantization/imagenet_gen_qsym.py
+++ b/example/quantization/imagenet_gen_qsym.py
@@ -155,6 +155,16 @@ if __name__ == '__main__':
if args.ctx == 'gpu':
calib_layer = lambda name: name.endswith('_output') and (name.find('conv') != -1
or name.find('fc') != -1)
+ excluded_sym_names += ['ch_concat_3a_chconcat',
+ 'ch_concat_3b_chconcat',
+ 'ch_concat_3c_chconcat',
+ 'ch_concat_4a_chconcat',
+ 'ch_concat_4b_chconcat',
+ 'ch_concat_4c_chconcat',
+ 'ch_concat_4d_chconcat',
+ 'ch_concat_4e_chconcat',
+ 'ch_concat_5a_chconcat',
+ 'ch_concat_5b_chconcat']
else:
calib_layer = lambda name: name.endswith('_output') and (name.find('conv') != -1)
excluded_sym_names += ['flatten', 'fc1']