initialize from darknet53

2018-10-30 14:58:56 +01:00 · 2018-10-30 14:58:56 +01:00 · ed0390d0b5
parent 332fe002b3
commit ed0390d0b5
1 changed files with 0 additions and 1 deletions
--- 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)