IndexError: index 0 is out of bounds for axis 0 with size 0
Closed this issue · 2 comments
A0401A commented
Do you have a great god who knows how to solve this problem?
File "F:\MemSeg-main\models\memory_module.py", line 33, in update
input_normal, _, _ = self.normal_dataset[samples_idx[i]]
~~~~~~~~~~~^^^
TooTouch commented
If you defined self.normal_dataset
as MemSegDataset
class in data folder, self.normal_dataset
should return img
, mask
, and target
as follows
Line 126 in 7e3a8b8
Did you use your custom dataset, not MemSegDataset
?
A0401A commented
Thank you for your reply. I am using MemSegDataset, and I added a dot before the dataset path to avoid any errors.