Skip to content

Adding a performance writer to log the loss on epoch validation end

Maxence Draguet requested to merge mdraguet/salt:max-addCallback-perf into main

This log is required in order to steer Katib when training on kubeflow. The log is set with the new PerformanceWriter.py callback and stored either in the same folder as the checkpoints or at a specified path in the base.yaml config (useful to grab easily in Katib).

Merge request reports