lijh1024

Results 1 issues of lijh1024

I use torch.nn.DataParallel() to split the model, but it can not split the input as it is a list of tensor