updates

2019-02-22 16:24:30 +01:00 · 2019-02-22 16:24:30 +01:00 · 9e60e97a6c
parent ac22a717f1
commit 9e60e97a6c
1 changed files with 1 additions and 1 deletions
--- 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'])