MO should support LRN k param with caffe model, rather than fixed to 1 (#716)
authorwistal <qypwl@163.com>
Wed, 3 Jun 2020 17:33:55 +0000 (01:33 +0800)
committerGitHub <noreply@github.com>
Wed, 3 Jun 2020 17:33:55 +0000 (20:33 +0300)
Co-authored-by: yipengqu <yipeng.qu@intel.com>
model-optimizer/extensions/front/caffe/lrn_ext.py

index a8ee3f59c6924e5465f07348296e3841dcc28c51..1c0a72ca06eec3157b094fc3370dc62fcd17b105 100644 (file)
@@ -29,7 +29,7 @@ class LRNExtractor(FrontExtractorOp):
         AttributedLRN.update_node_stat(node, {
             'alpha': param.alpha,
             'beta': param.beta,
-            'bias': 1,
+            'bias': param.k,
             'local_size': param.local_size,
             'region': region,
         })