Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-02-10 21:28:27 +01:00
parent c60bad8b10
commit 917f9dd248
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -52,7 +52,7 @@ def train(
model.load_state_dict(checkpoint['model']) model.load_state_dict(checkpoint['model'])
if torch.cuda.device_count() > 1: 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') # print('Using ', torch.cuda.device_count(), ' GPUs')
# model = nn.DataParallel(model) # model = nn.DataParallel(model)
model.to(device).train() model.to(device).train()