cellarium-ai / cellarium-ml

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

`LossScaleMonitor` #239

Closed ordabayevy closed 2 months ago

ordabayevy commented 2 months ago

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

ordabayevy commented 2 months ago

@sjfleming thanks for reviewing!