From 13fc171f60ae1981c7ad4f2a302a8a85c29addc5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=89tienne=20Simon?= Date: Thu, 23 Jul 2015 21:20:32 -0400 Subject: Use new tvt dataset with option --tvt --- train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'train.py') diff --git a/train.py b/train.py index 222f24c..cb9b63b 100755 --- a/train.py +++ b/train.py @@ -37,10 +37,10 @@ from ext_test import RunOnTest logger = logging.getLogger(__name__) if __name__ == "__main__": - if len(sys.argv) != 2: - print >> sys.stderr, 'Usage: %s config' % sys.argv[0] + if len(sys.argv) < 2 or len(sys.argv) > 3: + print >> sys.stderr, 'Usage: %s [--tvt | --largevalid] config' % sys.argv[0] sys.exit(1) - model_name = sys.argv[1] + model_name = sys.argv[-1] config = importlib.import_module('.%s' % model_name, 'config') logger.info('# Configuration: %s' % config.__name__) -- cgit v1.2.3