diff options
Diffstat (limited to 'sdil/data.py')
| -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. |