From 8370e766fbe1f88665f3bd85343378f1b096fbd8 Mon Sep 17 00:00:00 2001
From: YurenHao0426 <Blackhao0426@gmail.com>
Date: Thu, 6 Aug 2026 17:02:10 -0500
Subject: feat: record Rain run provenance

---
 experiments/rain_ep_bias_train.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'experiments')

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],
-- 
cgit v1.2.3