From 9e60e97a6cfd58fb7e91f256369dcc4ac838f39f Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 22 Feb 2019 16:24:30 +0100
Subject: [PATCH] updates

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

diff --git a/train.py b/train.py
index 6a7e6dc6..aa355c6f 100644
--- a/train.py
+++ b/train.py
@@ -43,7 +43,7 @@ def train(
     start_epoch = 0
     best_loss = float('inf')
     if resume:
-        checkpoint = torch.load('weights/yolov3.pt', map_location='cpu')
+        checkpoint = torch.load(latest, map_location='cpu')
 
         # Load weights to resume from
         model.load_state_dict(checkpoint['model'])