You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I am trying to run the model with BalancedDataParallel and I initialized my network with tt = TT(args, device).to(device); if ((not args.cpu) and (args.num_gpu > 1)): tt = BalancedDataParallel(1 ,tt, dim=0);.
But I keep getting this error: data_parallel_my.py", line 79, in scatter bsz = inputs[0].size(self.dim) IndexError: tuple index out of range
I printed inputs out and found it to be empty. My code works with DataParallel itself so I am not sure what went wrong. Here is where I try to call the network: srhd, S, T_lv3, T_lv2, T_lv1 = tt(lr=sr, ref=ref)
The text was updated successfully, but these errors were encountered:
I am trying to run the model with
BalancedDataParallel
and I initialized my network withtt = TT(args, device).to(device); if ((not args.cpu) and (args.num_gpu > 1)): tt = BalancedDataParallel(1 ,tt, dim=0);
.But I keep getting this error:
data_parallel_my.py", line 79, in scatter bsz = inputs[0].size(self.dim) IndexError: tuple index out of range
I printed
inputs
out and found it to be empty. My code works withDataParallel
itself so I am not sure what went wrong. Here is where I try to call the network:srhd, S, T_lv3, T_lv2, T_lv1 = tt(lr=sr, ref=ref)
The text was updated successfully, but these errors were encountered: