aimhubio / aim

Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.
https://aimstack.io
Apache License 2.0
5.16k stars 316 forks source link

[feat] Allow HF callback to initialize Aim run at init end #3116

Closed dushyantbehl closed 6 months ago

dushyantbehl commented 6 months ago

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.

dushyantbehl commented 6 months ago

@alberttorosyan sure!

dushyantbehl commented 6 months ago

@alberttorosyan done with the changes, let me know if I need to change anything. 🥂