model = nn.DataParallel(model, device_ids=[0, 1])
context, ctx_length = batch.context
response, rsp_length = batch.response
label = batch.label
prediction = self.model(context, response)
loss = self.criterion(prediction, label)
the batch size is 32, the size of prediction is 32 * 2 =  64, but the size of the label is still 32, which cause the criterion to raise Error.
What’s the problem?