What about a single tensor throwing the error?`
ipdb> o.shape
torch.Size([16, 46])
ipdb> o.device
device(type='cpu')
ipdb> device
device(type='cuda', index=0)
ipdb> o.to(device, non_blocking=True)
*** RuntimeError: CUDA error: device-side assert triggered
ipdb> list
245 # Cell
246 def to_device(b, device=None):
247 "Recursively put `b` on `device`."
248 if defaults.use_cuda==False: device='cpu'
249 elif device is None: device=default_device()
--> 250 def _inner(o): return o.to(device, non_blocking=True) if isinstance(o,Tensor) else o.to_device(device) if hasattr(o, "to_device") else o
251 return apply(_inner, b)
252
253 # Cell
254 def to_cpu(b):
255 "Recursively map lists of tensors in `b ` to the cpu."