From b6d40a34516b5441561340179c1d93e0370dd734 Mon Sep 17 00:00:00 2001 From: Josh Meyer Date: Wed, 21 Jul 2021 15:35:33 -0400 Subject: [PATCH] Add CI test for in-script variable setting --- bin/run-ldc93s1.py | 3 ++- ci_scripts/train-extra-tests.sh | 3 +++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/bin/run-ldc93s1.py b/bin/run-ldc93s1.py index d6103128..610d20ba 100755 --- a/bin/run-ldc93s1.py +++ b/bin/run-ldc93s1.py @@ -11,13 +11,14 @@ os.environ['CUDA_VISIBLE_DEVICES']='0' download_ldc("data/ldc93s1") initialize_globals_from_args( + load_train="init", alphabet_config_path="data/alphabet.txt", train_files=["data/ldc93s1/ldc93s1.csv"], dev_files=["data/ldc93s1/ldc93s1.csv"], test_files=["data/ldc93s1/ldc93s1.csv"], augment=["time_mask"], n_hidden=100, - epochs=100 + epochs=200 ) early_training_checks() diff --git a/ci_scripts/train-extra-tests.sh b/ci_scripts/train-extra-tests.sh index 1f76a0ed..50265afc 100755 --- a/ci_scripts/train-extra-tests.sh +++ b/ci_scripts/train-extra-tests.sh @@ -66,3 +66,6 @@ time ./bin/run-ci-ldc93s1_checkpoint_sdb.sh # Bytes output mode, resuming from checkpoint time ./bin/run-ci-ldc93s1_checkpoint_bytes.sh + +# Training with args set via initialize_globals_from_args() +time python ./bin/run-ldc93s1.py