Current implementation of HF callback in Aim initializes the Run object at training begin time. This means if the users want to track any additional metrics they need to do so before training finishes while using HFTrainer object.
This small change which initializes the Run object on init_end makes the run object available to user of this callback via callback.experiment and allows them to track any extra metrics or parameters with the same run other than the standard metrics output by HF.
Current implementation of HF callback in Aim initializes the
Run
object at training begin time. This means if the users want to track any additional metrics they need to do so before training finishes while usingHFTrainer
object.This small change which initializes the
Run
object oninit_end
makes the run object available to user of this callback viacallback.experiment
and allows them to track any extra metrics or parameters with the same run other than the standard metrics output by HF.