for ii , (X, y, y_weight) in enumerate(dataLoader[phase]): #for each of the batches
X = X.to(device) # [Nbatch, 3, H, W]
y_weight = y_weight.type(‘torch.FloatTensor’).to(device)
y = y.type(‘torch.LongTensor’).to(device)
error:
TypeError: self.dims,self.dims_chunk,self.maxdims cannot be converted to a Python object for pickling
====================================
for x,y,w in dataLoader[‘train’]:
print(x.shape, y.shape, w.shape)
error:
*RuntimeError: DataLoader worker (pid(s) 7744, 4584) exited unexpectedly