Chiaraplizz / ST-TR

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

IndexError: index 23667 is out of bounds for axis 0 with size 16560 #27

Closed jianghongcheng closed 3 years ago

jianghongcheng commented 3 years ago

Traceback (most recent call last): File "/media/max/b/2021/FG_2021/ST-TR-master_FIRST/src/main.py", line 973, in processor.start() File "/media/max/b/2021/FG_2021/ST-TR-master_FIRST/src/main.py", line 885, in start self.train(epoch, save_model=save_model) File "/media/max/b/2021/FG_2021/ST-TR-master_FIRST/src/main.py", line 529, in train for batch_idx, (data, label, name) in enumerate(loader): File "/home/max/anaconda3/envs/GaitGraph/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 517, in next data = self._next_data() File "/home/max/anaconda3/envs/GaitGraph/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1199, in _next_data return self._process_data(data) File "/home/max/anaconda3/envs/GaitGraph/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1225, in _process_data data.reraise() File "/home/max/anaconda3/envs/GaitGraph/lib/python3.7/site-packages/torch/_utils.py", line 429, in reraise raise self.exc_type(msg) IndexError: Caught IndexError in DataLoader worker process 0. Original Traceback (most recent call last): File "/home/max/anaconda3/envs/GaitGraph/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 202, in _worker_loop data = fetcher.fetch(index) File "/home/max/anaconda3/envs/GaitGraph/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/max/anaconda3/envs/GaitGraph/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/media/max/b/2021/FG_2021/ST-TR-master_FIRST/src/st_gcn/feeder/feeder.py", line 127, in getitem data_numpy = np.array(self.data[index]) File "/home/max/anaconda3/envs/GaitGraph/lib/python3.7/site-packages/numpy/core/memmap.py", line 331, in getitem res = super(memmap, self).getitem(index) IndexError: index 23667 is out of bounds for axis 0 with size 16560

jianghongcheng commented 3 years ago

need to keep same data with label