From 887ab29c64598d26fe334d0514678ba85d75e550 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 22 Nov 2018 20:03:09 +0100 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index e523d2d4..dbfd8749 100644 --- a/train.py +++ b/train.py @@ -18,7 +18,7 @@ opt = parser.parse_args() print(opt) # Import test.py to get mAP after each epoch -sys.argv[1:] = [] # delete any command line arguments that might get picked up by test.py +sys.argv[1:] = [] # delete any train.py command-line arguments before they reach test.py import test # must follow sys.argv[1:] = [] cuda = torch.cuda.is_available()