tensorflow / java

Java bindings for TensorFlow
Apache License 2.0
813 stars 200 forks source link

Adding a control dependency on the gradients to the gradient optimizers #520

Closed Craigacp closed 7 months ago

Craigacp commented 7 months ago

This improves determinism and makes the gradients be computed correctly. As it's a control dependency on all the gradients to the consumers of the gradients it's not clear why this is necessary, but it's working.

Fix developed by @nfeybesse.