GATorch: A PyTorch extension for energy consumption measurements

Date: Apr 01, 2023 Version: 0.0.2

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.

This guide provides the user with the explenention of GATorch concepts and functionality.

User Manual:

• Installation
  • Installing an official release
  • Verify Installation
• Getting started with GATorch
  • Retrieving the measurements
  • Tracking a model
  • Loss vs Energy Consumption
• Visualize Measurements
• Tensorboard Integration
  • Graphs
  • Overall Consumption

API Reference:

• API documentation
  • GATorch.GA