projects
/
platform
/
upstream
/
dldt.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
1ffada0
)
MO should support LRN k param with caffe model, rather than fixed to 1 (#716)
author
wistal
<qypwl@163.com>
Wed, 3 Jun 2020 17:33:55 +0000
(
01:33
+0800)
committer
GitHub
<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
patch
|
blob
|
history
diff --git
a/model-optimizer/extensions/front/caffe/lrn_ext.py
b/model-optimizer/extensions/front/caffe/lrn_ext.py
index a8ee3f59c6924e5465f07348296e3841dcc28c51..1c0a72ca06eec3157b094fc3370dc62fcd17b105 100644
(file)
--- a/
model-optimizer/extensions/front/caffe/lrn_ext.py
+++ b/
model-optimizer/extensions/front/caffe/lrn_ext.py
@@
-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,
})