for i, data in enumerate(train_dataloader, 0):
File “/home/vijay/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py”, line 637, in next
return self._process_next_batch(batch)
File “/home/vijay/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py”, line 657, in _process_next_batch
if isinstance(batch, ExceptionWrapper):            raise batch.exc_type(batch.exc_msg)
error