Nodule Detection Errors

[ghost]

Hi @ymli39 I am facing a starnge error when I am training the detector network. Do you have any comment on this error?

Traceback (most recent call last): File "/home/sh/Documents/Codes/My_Models/My_Seed/luna_detector/train_detector_se.py", line 351, in main() File "/home/sh/Documents/Codes/My_Models/My_Seed/luna_detector/train_detector_se.py", line 159, in main train(train_loader, net, loss, epoch, optimizer, get_lr, save_dir) File "/home/sh/Documents/Codes/My_Models/My_Seed/luna_detector/train_detector_se.py", line 187, in train for i, (data, target, coord) in enumerate(data_loader): File "/home/sh/Documents/Codes/DSB2017-master/venv/local/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 345, in next data = self._next_data() File "/home/sh/Documents/Codes/DSB2017-master/venv/local/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 856, in _next_data return self._process_data(data) File "/home/sh/Documents/Codes/DSB2017-master/venv/local/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 881, in _process_data data.reraise() File "/home/sh/Documents/Codes/DSB2017-master/venv/local/lib/python2.7/site-packages/torch/_utils.py", line 394, in reraise raise self.exc_type(msg) AssertionError: Caught AssertionError in DataLoader worker process 29. Original Traceback (most recent call last): File "/home/sh/Documents/Codes/DSB2017-master/venv/local/lib/python2.7/site-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop data = fetcher.fetch(index) File "/home/sh/Documents/Codes/DSB2017-master/venv/local/lib/python2.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch print(possibly_batched_index) File "/home/sh/Documents/Codes/My_Models/My_Seed/luna_detector/data_loader.py", line 108, in getitem label = self.label_mapping(sample.shape[1:], target, bboxes) File "/home/sh/Documents/Codes/My_Models/My_Seed/luna_detector/data_loader.py", line 286, in call assert (input_size[i] % stride == 0) AssertionError