diff options
| author | Matt Cooper <matthew_cooper@brown.edu> | 2016-08-12 20:36:02 -0400 |
|---|---|---|
| committer | Matt Cooper <matthew_cooper@brown.edu> | 2016-08-12 20:36:02 -0400 |
| commit | 856fc1973b0d63ef1841e37c3391dc49484e59a8 (patch) | |
| tree | 3c32c51ea8103a96ce303d4d868cc3c283b51a42 /Code/avg_runner.py | |
| parent | 42774cdcfec30be73e7ec6eedb82a8c53af4a857 (diff) | |
Made sure c.TEST_DIR exists before running
Diffstat (limited to 'Code/avg_runner.py')
| -rw-r--r-- | Code/avg_runner.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/Code/avg_runner.py b/Code/avg_runner.py index 5bf1e90..18bc592 100644 --- a/Code/avg_runner.py +++ b/Code/avg_runner.py @@ -1,6 +1,7 @@ import tensorflow as tf import getopt import sys +import os from utils import get_train_batch, get_test_batch import constants as c @@ -158,6 +159,10 @@ def main(): if opt == '--model_save_freq': c.MODEL_SAVE_FREQ = int(arg) + # set test frame dimensions + assert os.path.exists(c.TEST_DIR) + c.TEST_HEIGHT, c.TEST_WIDTH = c.get_test_frame_dims() + ## # Init and run the predictor ## |