Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-04-14 16:00:04 +02:00
parent 52464f5a06
commit 09949cdafa
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -101,7 +101,7 @@ def train(
mixed_precision = False mixed_precision = False
if mixed_precision: if mixed_precision:
from apex import amp from apex import amp
model, optimizer = amp.initialize(model, optimizer, opt_level='01') model, optimizer = amp.initialize(model, optimizer, opt_level='O1')
# Start training # Start training
t = time.time() t = time.time()