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)