Closed vcvishal closed 5 years ago
i am not sure where i am doing wrong would you please help me path is incorrect? image not found? or else?
Just wanted to make sure. Are you trying on DRIVE dataset or your own dataset?
yes i am using DRIVE
this time Total Params: 31042434
Override? [Y/N]: y
Override? [Y/N]: y
Override? [Y/N]: y
Patches: 135
Patches: 9
Patches: 9
Patches: 9
Patches: 9
Patches: 9
Training...
Traceback (most recent call last):
File "C:\Users\vcvis\Desktop\ature-master\testarch\unet__init.py", line 44, in run
epoch_run=drive_trainer.epoch_ce_loss)
File "C:\Users\vcvis\Desktop\ature-master\nbee\torchbee.py", line 82, in train
epoch_run(epoch=epoch, data_loader=data_loader)
File "C:\Users\vcvis\Desktop\ature-master\nbee\torchbee.py", line 210, in epoch_ce_loss
for i, data in enumerate(kw['data_loader'], 1):
File "D:\miniconda\lib\site-packages\torch\utils\data\dataloader.py", line 819, in iter__
return _DataLoaderIter(self)
File "D:\miniconda\lib\site-packages\torch\utils\data\dataloader.py", line 560, in init
w.start()
File "D:\miniconda\lib\multiprocessing\process.py", line 105, in start
self._popen = self._Popen(self)
File "D:\miniconda\lib\multiprocessing\context.py", line 223, in _Popen
return _default_context.get_context().Process._Popen(process_obj)
File "D:\miniconda\lib\multiprocessing\context.py", line 322, in _Popen
return Popen(process_obj)
File "D:\miniconda\lib\multiprocessing\popen_spawn_win32.py", line 65, in init
reduction.dump(process_obj, to_child)
File "D:\miniconda\lib\multiprocessing\reduction.py", line 60, in dump
ForkingPickler(file, protocol).dump(obj)
_pickle.PicklingError: Can't pickle <function
C:\Users\vcvis\Desktop\ature-master>Total Params: 31042434
Traceback (most recent call last):
File "
error might be due to multi threading issue where should i modifier for it
thank you problem solved by setting number of workers =0
Awesome, Thanks for pointing out. I never ran in CPU so hadn't encounter it before.
when I run this code the following output is given C:\Users\HI\Desktop\ature-master>python main.py Total Params: 31042434 [0, 0, 0, 0] what is it, how I get the correct output I am using windows