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

initialize from darknet53

This commit is contained in:
Glenn Jocher 2018-10-30 14:58:56 +01:00
parent 332fe002b3
commit ed0390d0b5
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

@ -74,7 +74,6 @@ def main(opt):
del checkpoint # current, saved del checkpoint # current, saved
else: else:
load_weights(model, 'weights/darknet53.conv.74') # load darknet53 weights (optional) load_weights(model, 'weights/darknet53.conv.74') # load darknet53 weights (optional)
if torch.cuda.device_count() > 1: if torch.cuda.device_count() > 1:
print('Using ', torch.cuda.device_count(), ' GPUs') print('Using ', torch.cuda.device_count(), ' GPUs')
model = nn.DataParallel(model) model = nn.DataParallel(model)