From b025b3123e32d694bf0b493afa026deb2a77e76c Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 23 Jul 2019 15:08:28 +0200
Subject: [PATCH] updates

---
 train.py             | 2 +-
 utils/torch_utils.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 38f66efa..1d08a439 100644
--- a/train.py
+++ b/train.py
@@ -138,7 +138,7 @@ def train(cfg,
     # lf = lambda x: 10 ** (hyp['lrf'] * x / epochs)  # exp ramp
     # lf = lambda x: 1 - 10 ** (hyp['lrf'] * (1 - x / epochs))  # inverse exp ramp
     # scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lf)
-    scheduler = lr_scheduler.MultiStepLR(optimizer, milestones=[round(opt.epochs * x) for x in (0.8, 0.9)], gamma=0.1)
+    scheduler = lr_scheduler.MultiStepLR(optimizer, milestones=[round(opt.epochs * x) for x in [0.8]], gamma=0.1)
     scheduler.last_epoch = start_epoch - 1
 
     # # Plot lr schedule
diff --git a/utils/torch_utils.py b/utils/torch_utils.py
index adbc8705..b61050fa 100644
--- a/utils/torch_utils.py
+++ b/utils/torch_utils.py
@@ -18,7 +18,7 @@ def select_device(force_cpu=False):
     if cuda:
         try:  # Mixed precision training https://github.com/NVIDIA/apex
             from apex import amp
-            apex_str = 'with Apex '
+            apex_str = 'Apex '
         except:
             apex_str = ''