From 0caca675d322362a96d18b436228466d790cd945 Mon Sep 17 00:00:00 2001
From: Lev Zlotnik <46742999+levzlotnik@users.noreply.github.com>
Date: Mon, 9 Sep 2019 17:53:08 +0300
Subject: [PATCH] Bug fix in PTQ yaml
---
.../resnet18_imagenet_post_train.yaml | 14 ++++++++------
1 file changed, 8 insertions(+), 6 deletions(-)
diff --git a/examples/quantization/post_train_quant/resnet18_imagenet_post_train.yaml b/examples/quantization/post_train_quant/resnet18_imagenet_post_train.yaml
index ac74ef6..cf91160 100644
--- a/examples/quantization/post_train_quant/resnet18_imagenet_post_train.yaml
+++ b/examples/quantization/post_train_quant/resnet18_imagenet_post_train.yaml
@@ -67,16 +67,18 @@ quantizers:
conv1:
# The input to the first layer in the model will never have quantization metadata
input_overrides:
- # Shorthand to take the quantization settings of the output (ignores any other settings)
- from_outputs: True
+ 0:
+ # Shorthand to take the quantization settings of the output (ignores any other settings)
+ from_output: True
fc:
# In ResNet, the FC layer has a view op before, which kills the quantization metadata. So we have to override
# (or enable auto_fallback).
input_overrides:
- # Example of setting the actual value. Applicable only if 'from_outputs' isn't set.
- # The following keys are supported: 'bits_activations', 'mode', 'clip_acts', 'clip_n_stds'
- # Any key not explicitly set will default to the output setting
- bits_activations: 6
+ 0:
+ # Example of setting the actual value. Applicable only if 'from_outputs' isn't set.
+ # The following keys are supported: 'bits_activations', 'mode', 'clip_acts', 'clip_n_stds'
+ # Any key not explicitly set will default to the output setting
+ bits_activations: 6
# Overrides section for run 3
# overrides:
--
GitLab