Scale mismatch when using average pooling.

intel / he-transformer

nGraph-HE: Deep learning with Homomorphic Encryption (HE) through Intel nGraph

Apache License 2.0

172 stars 35 forks source link

Scale mismatch when using average pooling. #61

Closed Wixee closed 3 years ago

Wixee commented 3 years ago

I meet scale mismatch error when using average pooling. But this issue has been fixed in https://github.com/NervanaSystems/he-transformer/issues/229, but not in IntelAI/he-transformer. And I find bug #227 in NervanaSystems/he-transformer, are not fixed in IntelAI/he-transformer too, and I think there will be more issues like these. @fboemer