diff options
| author | YurenHao0426 <Blackhao0426@gmail.com> | 2026-07-30 16:34:30 -0500 |
|---|---|---|
| committer | YurenHao0426 <Blackhao0426@gmail.com> | 2026-07-30 16:34:30 -0500 |
| commit | 716205eba862076c1d0fbbff016601272e4614f2 (patch) | |
| tree | dc8c31dcfefa49d53c98d305fe35419da76e58c9 /sdil | |
| parent | 906d17e896329e4d2dd8d7f728e4d5a234afe0e4 (diff) | |
experiment: package portable A6000 crossover
Diffstat (limited to 'sdil')
| -rw-r--r-- | sdil/data.py | 3 |
1 files changed, 2 insertions, 1 deletions
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. |