feat: record Rain run provenance

1 files changed, 10 insertions, 0 deletions

diff --git a/experiments/rain_ep_bias_train.py b/experiments/rain_ep_bias_train.py
index abc4138..80135ea 100644
--- a/experiments/rain_ep_bias_train.py
+++ b/experiments/rain_ep_bias_train.py
@@ -5,6 +5,7 @@ from __future__ import annotations
 
 import argparse
 import json
+import os
 from pathlib import Path
 import random
 import subprocess
@@ -254,6 +255,7 @@ def main() -> None:
 
     report = {
         "schema": "rain_ep_structured_bias_screen_v1",
+        "sdil": {"revision": revision(ROOT)},
         "author": {
             "repository": "https://github.com/rain-neuromorphics/energy-based-learning",
             "revision": author_revision,
@@ -294,6 +296,14 @@ def main() -> None:
                 "best_effort_warn_only" if args.deterministic else "author_default"),
             "autodiff_used_for_learning": False,
         },
+        "hardware": {
+            "torch_version": torch.__version__,
+            "torch_cuda_version": torch.version.cuda,
+            "cuda_visible_devices": os.environ.get("CUDA_VISIBLE_DEVICES"),
+            "device_name": (
+                torch.cuda.get_device_name(device)
+                if device.type == "cuda" else "cpu"),
+        },
         "metrics": metrics,
         "epochs_completed": len(metrics),
         "final": metrics[-1],