Hugging Face Trainer callback integration

ml-energy / zeus

Deep Learning Energy Measurement and Optimization

https://ml.energy/zeus

Apache License 2.0

180 stars 24 forks source link

Hugging Face Trainer callback integration #24

Closed jaywonchung closed 4 months ago

jaywonchung commented 9 months ago

It would be convenient to just have an in-tree implementation for Hugging Face callbacks so that people can just pop it in Trainer's callbacks parameter and get energy measurement and optimization for free.