@ptrblck  Thank you very much for your reply.



 ptrblck:

nn.DataParallel will chunk the input data in dim0, so you should make sure to permute your data and adapt the model if necessary.


Is dim0 the dimension of batch？ If so, the inputs in my code no longer need to be permuted, right? because ‘100’ in dim0 indicates the batch_size.



 marathon:

inputs = torch.zeros(100, 64, 256).cuda()