This commit is contained in:
Glenn Jocher 2019-05-23 12:36:13 +02:00
parent 3006c33c29
commit 68b9df4dd4
1 changed files with 1 additions and 1 deletions

View File

@ -168,7 +168,7 @@ def train(
model, optimizer = amp.initialize(model, optimizer, opt_level='O1') model, optimizer = amp.initialize(model, optimizer, opt_level='O1')
# Remove old results # 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) os.remove(f)
# Start training # Start training