diff options
| -rw-r--r-- | experiments/rain_ep_bias_train.py | 10 |
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], |