updates
This commit is contained in:
parent
3006c33c29
commit
68b9df4dd4
2
train.py
2
train.py
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue