From ff06b698ed76d1bb10733ec6de2813f777f2a258 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 25 Mar 2019 15:32:00 +0100
Subject: [PATCH] updates

---
 train.py                           | 2 --
 weights/download_yolov3_weights.sh | 1 -
 2 files changed, 3 deletions(-)

diff --git a/train.py b/train.py
index 53f08d71..4611405e 100644
--- a/train.py
+++ b/train.py
@@ -190,8 +190,6 @@ def train(
                 os.system('cp ' + latest + ' ' + weights + 'backup%g.pt' % epoch)
 
         # Calculate mAP
-        if type(model) is nn.parallel.DistributedDataParallel:
-            model = model.module
         with torch.no_grad():
             P, R, mAP = test.test(cfg, data_cfg, weights=latest, batch_size=batch_size, img_size=img_size)
 
diff --git a/weights/download_yolov3_weights.sh b/weights/download_yolov3_weights.sh
index fe6213aa..0568cb87 100644
--- a/weights/download_yolov3_weights.sh
+++ b/weights/download_yolov3_weights.sh
@@ -18,4 +18,3 @@ wget -c https://pjreddie.com/media/files/darknet53.conv.74
 # ./darknet partial cfg/yolov3-tiny.cfg yolov3-tiny.weights yolov3-tiny.conv.15 15
 # mv yolov3-tiny.conv.15 ../
 
-cd ..