search

Chiaraplizz / ST-TR

Spatial Temporal Transformer Network for Skeleton-Based Activity Recognition
MIT License
299 stars 58 forks source link

Hii, Chiaraplize, I had a DataLoader problem while running the code. #33

Open Kapil-Bhatt opened 2 years ago

Kapil-Bhatt commented 2 years ago

Traceback (most recent call last): File "/content/drive/MyDrive/ST-TR-master/code/main.py", line 974, in processor.start() File "/content/drive/MyDrive/ST-TR-master/code/main.py", line 886, in start self.train(epoch, save_model=save_model) File "/content/drive/MyDrive/ST-TR-master/code/main.py", line 442, in train for batch_idx, (data, label, name) in enumerate(loader): File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 521, in next data = self._next_data() File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 1203, in _next_data return self._process_data(data) File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 1229, in _process_data data.reraise() File "/usr/local/lib/python3.7/dist-packages/torch/_utils.py", line 434, in reraise raise exception AttributeError: Caught AttributeError in DataLoader worker process 0.

I hope you can help me to solve this problem.