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-18 16:45:38 +02:00
parent c68093ee7d
commit 05ec8b3b9d
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

@ -120,6 +120,7 @@ def train(
sampler=sampler) sampler=sampler)
# Mixed precision training https://github.com/NVIDIA/apex # Mixed precision training https://github.com/NVIDIA/apex
# install help: https://github.com/NVIDIA/apex/issues/259
mixed_precision = False mixed_precision = False
if mixed_precision: if mixed_precision:
from apex import amp from apex import amp