From 09949cdafa88fd0d7fa85f776fa5b7ebf8bc6dbb Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 14 Apr 2019 16:00:04 +0200
Subject: [PATCH] updates

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index bb1498da..ba134bc4 100644
--- a/train.py
+++ b/train.py
@@ -101,7 +101,7 @@ def train(
     mixed_precision = False
     if mixed_precision:
         from apex import amp
-        model, optimizer = amp.initialize(model, optimizer, opt_level='01')
+        model, optimizer = amp.initialize(model, optimizer, opt_level='O1')
 
     # Start training
     t = time.time()