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'])