Set torch seeds for reproducible training

alexandertuna · alexandertuna · commit a9ed4ed03560 · 2025-06-27T13:36:32.000-07:00
diff --git a/RecoTracker/LSTCore/standalone/analysis/DNN/train_T3_DNN.ipynb b/RecoTracker/LSTCore/standalone/analysis/DNN/train_T3_DNN.ipynb
@@ -1,5 +1,16 @@
 {
   "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# set seed for reproducibility\n",
+        "import torch\n",
+        "torch.manual_seed(42)"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": 1,
diff --git a/RecoTracker/LSTCore/standalone/analysis/DNN/train_T5_DNN.ipynb b/RecoTracker/LSTCore/standalone/analysis/DNN/train_T5_DNN.ipynb
@@ -1,5 +1,16 @@
 {
  "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# set seed for reproducibility\n",
+    "import torch\n",
+    "torch.manual_seed(42)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 1,
diff --git a/RecoTracker/LSTCore/standalone/analysis/DNN/train_pT3_DNN.ipynb b/RecoTracker/LSTCore/standalone/analysis/DNN/train_pT3_DNN.ipynb
@@ -1,5 +1,16 @@
 {
   "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# set seed for reproducibility\n",
+        "import torch\n",
+        "torch.manual_seed(42)"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": 1,
