Merge pull request #617 from take-cheeze/opt_disable_per_channel_quan…

…tize Add --disable_per_channel_quantize option
pfnet-research · Aug 27, 2019 · b8ba8b5 · b8ba8b5
2 parents 09c85e3 + eeb6612
commit b8ba8b5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/compiler/passes.cc b/compiler/passes.cc
@@ -118,6 +118,7 @@ void RunDefaultPasses(Graph* graph, bool gen_backprop, bool skip_scheduling) {
 
         if (g_quantize) {
             QuantizationOptions q_opts;
+            q_opts.per_channel = !g_disable_per_channel_quantize;
             Recursively([q_opts](Graph* graph) { Quantize(q_opts, graph); }, graph);
         }
 

diff --git a/scripts/generate_flags_code.py b/scripts/generate_flags_code.py
@@ -118,6 +118,10 @@
         'type': 'bool',
         'doc': 'Quantize ONNX model'
     },
+    'disable_per_channel_quantize': {
+        'type': 'bool',
+        'doc': 'Disables per channel quantization'
+    },
 
     'computation_order': {
         'type': 'std::string',