updates
This commit is contained in:
parent
e82f201578
commit
56f38ed6a2
4
train.py
4
train.py
|
@ -14,8 +14,8 @@ from utils.utils import *
|
||||||
mixed_precision = True
|
mixed_precision = True
|
||||||
try: # Mixed precision training https://github.com/NVIDIA/apex
|
try: # Mixed precision training https://github.com/NVIDIA/apex
|
||||||
from apex import amp
|
from apex import amp
|
||||||
except: # not installed: install help: https://github.com/NVIDIA/apex/issues/259
|
except:
|
||||||
mixed_precision = False
|
mixed_precision = False # not installed
|
||||||
|
|
||||||
# 320 --epochs 1
|
# 320 --epochs 1
|
||||||
# 0.109 0.297 0.150 0.126 7.04 1.666 4.062 0.1845 42.6 3.34 12.61 8.338 0.2705 0.001 -4 0.9 0.0005 a 320 giou + best_anchor False
|
# 0.109 0.297 0.150 0.126 7.04 1.666 4.062 0.1845 42.6 3.34 12.61 8.338 0.2705 0.001 -4 0.9 0.0005 a 320 giou + best_anchor False
|
||||||
|
|
Loading…
Reference in New Issue