You mean
model = nn.DataParallel(model)  # The order is wrong?
model.to(device)

model.to(device)  # This is right?
model = nn.DataParallel(model)

But in the doc
image891×342 35.2 KB
BTW, where is the complete doc of nn.DataParallel??