From 917f9dd24823c1db29a604e1cca282e198fee0f0 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 10 Feb 2019 21:28:27 +0100 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 5793a322..69eb7bf7 100644 --- a/train.py +++ b/train.py @@ -52,7 +52,7 @@ def train( model.load_state_dict(checkpoint['model']) if torch.cuda.device_count() > 1: - raise Exception('Multi-GPU not currently supported: https://github.com/ultralytics/yolov3/issues/21') + raise Exception('Multi-GPU issue: https://github.com/ultralytics/yolov3/issues/21') # print('Using ', torch.cuda.device_count(), ' GPUs') # model = nn.DataParallel(model) model.to(device).train()