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
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