logangilmour / foveatedpyramid Goto Github PK

Hi,
Thank you for sharing your codes.

When I run your codes, however I got several warnings and errors. Can you help this out?
Also, can you provide your detail experiment enviroment? Thank you

Warnings
train.py:137: UserWarning: Using a target size (torch.Size([2, 1, 2])) that is different to the input size (torch.Size([2, 2, 2])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size. loss = F.mse_loss(outputs, labels_tensor,reduction='none') train.py:143: UserWarning: Using a target size (torch.Size([2, 1, 2])) that is different to the input size (torch.Size([2, 2, 2])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size. F.l1_loss(outputs, labels_tensor, reduction='mean').backward()

Error
Traceback (most recent call last): File "train.py", line 269, in <module> train("test_avg_up",[0],num_folds=2,fold_size=150,fold=1,graphical=False,avg_labels=True,iterations=2,batched=True)#,load=True,startEpoch=20) File "train.py", line 93, in train next_batch = data_iter.next() # start loading the first batch File "/home/x/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 435, in __next__ data = self._next_data() File "/home/x/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1085, in _next_data return self._process_data(data) File "/home/x/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1111, in _process_data data.reraise() File "/home/x/lib/python3.7/site-packages/torch/_utils.py", line 428, in reraise raise self.exc_type(msg) IndexError: Caught IndexError in DataLoader worker process 0. Original Traceback (most recent call last): File "/home/x/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 198, in _worker_loop data = fetcher.fetch(index) File "/home/x/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/x/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp> data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/x/FoveatedPyramid/XrayData.py", line 71, in __getitem__ x, junior, senior = self.dataset[self.indices[idx]] File "/home/x/FoveatedPyramid/XrayData.py", line 42, in __getitem__ path, junior, senior = self.files[idx] IndexError: index 150 is out of bounds for axis 0 with size 150