fix: Log trainer callback metrics to run (ENG-1911)

[monoxgas]

• Updated metric logging calls in DreadnodeCallback to include to="run" parameter.
• Changed the following logging statements:
  • dn.log_metric("epoch", state.epoch) to dn.log_metric("epoch", state.epoch, to="run")
  • dn.log_metric("step", state.global_step) to dn.log_metric("step", state.global_step, to="run")
  • Added to="run" to dynamic metric logging for cleaned keys.

---

Generated Summary:

• Updated metric logging calls in DreadnodeCallback to consistently include the to="run" parameter.
• Changes made to the following methods:
  • log_metric for epoch in the method handling epoch logging.
  • log_metric for step in the method handling step logging.
  • log_metric in the method iterating through cleaned keys in logs.
• These modifications ensure that metrics are correctly associated with the current run, potentially improving tracking and debugging capabilities.

This summary was generated with ❤️ by rigging