Add train_single.py for testing energy monitoring metrics

ml-energy · Nov 11, 2024 · e8bfe7b · e8bfe7b
1 parent 4681796
commit e8bfe7b
Show file tree

Hide file tree

Showing 2 changed files with 411 additions and 0 deletions.
diff --git a/examples/prometheus/requirements.txt b/examples/prometheus/requirements.txt
@@ -0,0 +1,2 @@
+torch
+torchvision