GATorch

Energy Tracker

Tools to track and analyze energy consumption of machine learning models during training

GATorch is a tool seamlessly integrated with PyTorch that enables ML developers to generate an energy consumption report. By attaching your model, the tool automatically tracks the energy consumption of your model's training and generates graphs and plots to gain in-depth insights into the energy consumption of your model.

GitHub

9 stars
2 watching
1 forks
Language: Python
last commit: almost 2 years ago
energy-consumptiongreen-aipytorchsustainability