Skip to content
This repository was archived by the owner on Oct 13, 2021. It is now read-only.

fused batchnorm result in conv #710

Open
pilz22 opened this issue Apr 27, 2021 · 1 comment
Open

fused batchnorm result in conv #710

pilz22 opened this issue Apr 27, 2021 · 1 comment

Comments

@pilz22
Copy link

pilz22 commented Apr 27, 2021

hello,
here I first tried to convert my keras model to onnx format. when I check the nodes in graph,it seems all conv2d and batchnorm are merged in to one node, the optype is 'conv'. and then i have to convert the onnx model to other model type. I want to consult whether and how the merged conv keeps the functions of batchnorm? if it is only a conv, then the batchnorm layer is missed when converted to other model type?

@jiafatom
Copy link
Collaborator

jiafatom commented Apr 28, 2021

We use ConvBatchNormOptimizer in onnxconverter-common to fuse conv+batchnorm, see the line here

# for free to subscribe to this conversation on GitHub. Already have an account? #.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants