cellarium-ai / cellarium-ml

Distributed single-cell data analysis.
BSD 3-Clause "New" or "Revised" License
20 stars 3 forks source link

`LossScaleMonitor` #239

Closed ordabayevy closed 1 month ago

ordabayevy commented 1 month ago

A callback that logs the loss scale during mixed-precision training.

ordabayevy commented 1 month ago

@sjfleming thanks for reviewing!