runing the “train.sh”,I met this question, could you tell me how to solve it? please.
Traceback (most recent call last):
File "training/main.py", line 528, in
main()
File "training/main.py", line 264, in main
train(train_loader, net, loss, epoch, optimizer, get_lr, args.save_freq, save_dir)
File "training/main.py", line 284, in train
for i, (data, target, coord) in enumerate(data_loader):
File "/usr/local/lib/python2.7/dist-packages/torch/utils/data/dataloader.py", line 179, in next
batch = self.collate_fn([self.dataset[i] for i in indices])
File "/media/dai/Data/Data/WL/work/lung_nodule_detector-master/training/data.py", line 95, in getitem
label = self.label_mapping(sample.shape[1:], target, bboxes)
File "/media/dai/Data/Data/WL/work/lung_nodule_detector-master/training/data.py", line 284, in call
assert(input_size[i] % stride == 0)
AssertionError
runing the “train.sh”,I met this question, could you tell me how to solve it? please.
Traceback (most recent call last):
File "training/main.py", line 528, in
main()
File "training/main.py", line 264, in main
train(train_loader, net, loss, epoch, optimizer, get_lr, args.save_freq, save_dir)
File "training/main.py", line 284, in train
for i, (data, target, coord) in enumerate(data_loader):
File "/usr/local/lib/python2.7/dist-packages/torch/utils/data/dataloader.py", line 179, in next
batch = self.collate_fn([self.dataset[i] for i in indices])
File "/media/dai/Data/Data/WL/work/lung_nodule_detector-master/training/data.py", line 95, in getitem
label = self.label_mapping(sample.shape[1:], target, bboxes)
File "/media/dai/Data/Data/WL/work/lung_nodule_detector-master/training/data.py", line 284, in call
assert(input_size[i] % stride == 0)
AssertionError