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

updates

This commit is contained in:
Glenn Jocher 2019-07-16 18:49:54 +02:00
parent 51d7e460a3
commit 64b606a3cd
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

@ -156,6 +156,7 @@ def train(cfg,
try: try:
from apex import amp from apex import amp
model, optimizer = amp.initialize(model, optimizer, opt_level='O1', verbosity=0) model, optimizer = amp.initialize(model, optimizer, opt_level='O1', verbosity=0)
print('Using Apex')
except: # not installed: install help: https://github.com/NVIDIA/apex/issues/259 except: # not installed: install help: https://github.com/NVIDIA/apex/issues/259
mixed_precision = False mixed_precision = False