Mentions légales du service

Skip to content

Improve Model and its subclasses.

ANDREY Paul requested to merge optimize-model into main

Clean-up Model and its subclasses:

  • Improve documentation of Model methods.
  • Harmonize _unpack_batch method name for the current Model subclasses.
  • Harmonize loss-averaging behavior of the current Model subclasses.

Improve TensorflowModel:

  • Optimize gradients computation in TensorflowModel using tf.function.
  • Improve the management and serialization of loss in TensorflowModel.
  • Implement utilities to build keras Loss objects from strings or functions.

Merge request reports