From 741626c55bbbc13436a2e49e8b3e1dbd5711fbd0 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 30 Oct 2018 15:20:52 +0100 Subject: [PATCH] initialize from darknet53 --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 3236786d..9f9798f9 100644 --- a/train.py +++ b/train.py @@ -76,7 +76,7 @@ def main(opt): else: # Initialize model with darknet53 weights (optional) if not os.path.isfile('weights/darknet53.conv.74'): - os.system('wget https://pjreddie.com/media/files/darknet53.conv.74 -P /weights') + os.system('wget https://pjreddie.com/media/files/darknet53.conv.74 -P weights') load_weights(model, 'weights/darknet53.conv.74') if torch.cuda.device_count() > 1: