search

lfz / DSB2017

The solution of team 'grt123' in DSB2017
MIT License
1.24k stars 418 forks source link

Run main.py for training error #15

Open aguang1201 opened 7 years ago

aguang1201 commented 7 years ago

I had finishing prepare datas.When I train with prepared DSB3 data and LUNA16 data,it is something wrong with DataLoader. using gpu 0,1 Traceback (most recent call last): File "/home/wisdom/soft/pycharm-community-2017.1/helpers/pydev/pydevd.py", line 1578, in globals = debugger.run(setup['file'], None, None, is_module) File "/home/wisdom/soft/pycharm-community-2017.1/helpers/pydev/pydevd.py", line 1015, in run pydev_imports.execfile(file, globals, locals) # execute the script File "/home/wisdom/PycharmProjects/DSB2017-master/training/detector/main.py", line 348, in main() File "/home/wisdom/PycharmProjects/DSB2017-master/training/detector/main.py", line 167, in main train(train_loader, net, loss, epoch, optimizer, get_lr, args.save_freq, save_dir) File "/home/wisdom/PycharmProjects/DSB2017-master/training/detector/main.py", line 179, in train for i, (data, target, coord) in enumerate(data_loader): File "/home/wisdom/anaconda3/envs/DSB2017/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 198, in next return self._process_next_batch(batch) File "/home/wisdom/anaconda3/envs/DSB2017/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 239, in _process_next_batch raise batch.exc_type(batch.exc_msg) TypeError: Traceback (most recent call last): File "/home/wisdom/anaconda3/envs/DSB2017/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 41, in _worker_loop samples = collate_fn([dataset[i] for i in batch_indices]) File "/home/wisdom/anaconda3/envs/DSB2017/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 109, in default_collate transposed = zip(*batch) TypeError: zip argument #3 must support iteration

I had modified the GPU=1,2 and workers=10 Thank you!