diff --git a/train.py b/train.py
index 53f08d7..4611405 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 fe6213a..0568cb8 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 ..