Add BatchNorm support
Nobody has claimed this yet.
- Dominant language
- Java
- Stars
- 928
- Forks
- 227
- PR merge metrics
- No merged PRs in 30d
Description
System information
- 1.15 and 2.3:
- Are you willing to contribute it (Yes):
Describe the feature and the current behavior/state.
The problem is described in the mailing thread.
I have next problem: trying to repeat one of the modern CNN architectures on Java API. Most of them are using BatchNormalization as a popular layer with tf.nn.batchNormalization() op.
I trying to use old operands like BatchNormWithGlobalNormalization
I've used but got Exception in thread "main" org.tensorflow.exceptions.TFUnimplementedException: Op BatchNormWithGlobalNormalization is not available in GraphDef version 175. It has been removed in version 9. Use tf.nn.batch_normalization().
at org.tensorflow.internal.c_api.AbstractTF_Status.throwExceptionIfNotOK(AbstractTF_Status.java:99)
This was deperecated years ago, but we have it in 1.15 and 2.x APIs.
The possible solutions:
- To implement all required stuff in Java Ops
- Add gradients and contributions in TF Core (gradients and another stuff) if it's possible
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reading the linked mailing thread and comparing the Java API's current support with tf.nn.batchNormalization(). The issue proposes either implementing the required Java Ops or adding TensorFlow Core gradients and related contributions, so the expected implementation path and completion criteria need to be clarified first.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- java
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 20/100