From ed0390d0b5997be626b65f5e74a0dfacc7037a36 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 30 Oct 2018 14:58:56 +0100 Subject: [PATCH] initialize from darknet53 --- train.py | 1 - 1 file changed, 1 deletion(-) diff --git a/train.py b/train.py index fe9f56e1..cbee6a75 100644 --- a/train.py +++ b/train.py @@ -74,7 +74,6 @@ def main(opt): del checkpoint # current, saved else: load_weights(model, 'weights/darknet53.conv.74') # load darknet53 weights (optional) - if torch.cuda.device_count() > 1: print('Using ', torch.cuda.device_count(), ' GPUs') model = nn.DataParallel(model)