From 68b9df4dd494262a42c2432792b3cbe8b0d6a61a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 23 May 2019 12:36:13 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 0ab05932..5a1e0cd4 100644 --- a/train.py +++ b/train.py @@ -168,7 +168,7 @@ def train( model, optimizer = amp.initialize(model, optimizer, opt_level='O1') # Remove old results - for f in glob.glob('train_batch*.jpg') + glob.glob('test_batch*.jpg') + 'results.txt': + for f in glob.glob('train_batch*.jpg') + glob.glob('test_batch*.jpg') + ['results.txt']: os.remove(f) # Start training