tensorflow / java

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

Reductions on losses that have dynamic size #523

Closed nfeybesse closed 7 months ago

nfeybesse commented 7 months ago

Currently, AUTO and SUM_OVER_BATCH reduction operations on all losses fail when the batch size is dynamically defined. These reduction operations are very frequently used. Dynamic definition of batch size as well.

The problem comes from the LossesHelper class which seeks to divide the sum of the error by the size of the batch which is not dynamically obtained currently