Skip to content
This repository has been 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

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

No branches or pull requests

2 participants