Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] Add BatchNorm layer #34

Closed
zaleslaw opened this issue Jan 21, 2021 · 1 comment
Closed

[Feature] Add BatchNorm layer #34

zaleslaw opened this issue Jan 21, 2021 · 1 comment
Assignees
Labels
enhancement New feature or request
Milestone

Comments

@zaleslaw
Copy link
Collaborator

zaleslaw commented Jan 21, 2021

Keras framework has a BatchNorm layer, and it's a significant and widely used layer in modern CNN architectures.

Unfortunately, the TF Java API 1.15 and TF Java API 2.x wraps outdated BatchNorm operand; alternative FusedBatchNorm is a real operand at all and could not be used as part of the trained graph.

The best option here creates our own BatchNorm operand based on another low-level TF operation with the ability to participate in backward and forward calculations.

@zaleslaw zaleslaw added the enhancement New feature or request label Jan 21, 2021
@zaleslaw zaleslaw added this to the 0.2 milestone Jan 21, 2021
@zaleslaw zaleslaw self-assigned this Jan 21, 2021
@zaleslaw zaleslaw changed the title [Feature Request] Add BatchNorm layer [Feature] Add BatchNorm layer Jan 21, 2021
@zaleslaw
Copy link
Collaborator Author

Unfortunately, FusedBatchNorm doesn't have gradients support in TF C/C++ API, the basic implementation for working in inference mode is implemented and merged.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant