diff options
Diffstat (limited to 'experiments/transformer_crossover_native.py')
| -rw-r--r-- | experiments/transformer_crossover_native.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/experiments/transformer_crossover_native.py b/experiments/transformer_crossover_native.py index ca47751..6a0269e 100644 --- a/experiments/transformer_crossover_native.py +++ b/experiments/transformer_crossover_native.py @@ -25,7 +25,10 @@ METHODS = ( "bp", "fa", "dfa", "pepita", "ff", "ep", "dualprop", "clean_kp", "sdil", ) -DEFAULT_DATA_DIR = "/home/yurenh2/local-transformer/data/shakespeare_char" +DEFAULT_DATA_DIR = os.environ.get( + "SDIL_TRANSFORMER_DATA_DIR", + "/home/yurenh2/local-transformer/data/shakespeare_char", +) def file_sha256(path): |