From 716205eba862076c1d0fbbff016601272e4614f2 Mon Sep 17 00:00:00 2001
From: YurenHao0426 <Blackhao0426@gmail.com>
Date: Thu, 30 Jul 2026 16:34:30 -0500
Subject: experiment: package portable A6000 crossover

---
 sdil/data.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'sdil')

diff --git a/sdil/data.py b/sdil/data.py
index 7f49b4e..759d7f7 100644
--- a/sdil/data.py
+++ b/sdil/data.py
@@ -2,13 +2,14 @@
 so it runs on the Pascal env on the 1080 farm). Datasets already live on the
 shared NFS home; the box is offline for dataset mirrors."""
 
+import os
 import pickle
 import struct
 import hashlib
 import numpy as np
 import torch
 
-DATA_DIR = "/home/yurenh2/sdrn/data"
+DATA_DIR = os.environ.get("SDIL_DATA_DIR", "/home/yurenh2/sdrn/data")
 
 # (subdir, mean, std) matching the standard torchvision normalisation used by
 # the predecessor project, so numbers are comparable.
-- 
cgit v1.2.3