ethnhe / PVN3D

Code for "PVN3D: A Deep Point-wise 3D Keypoints Hough Voting Network for 6DoF Pose Estimation", CVPR 2020
MIT License
492 stars 105 forks source link

TypeError: can't pickle module objects #120

Closed aghaphd closed 1 year ago

aghaphd commented 1 year ago

I'm running the PVN3D model on Win64-11, using the Anaconda Cuda 9.0. Py torch 1.0.1 as required. Facing a problem regarding the PCL library as I believe:

666@Ali MINGW64 /d/PVN3D-master/PVN3D-master/pvn3d $ python -m train.train_linemod_pvn3d --cls ape cls_type: ape cls_id in lm_dataset.py 1 train_dataset_size: 2186 cls_id in lm_dataset.py 1 val_dataset_size: 1050 loading pretrained mdl. C:\Users\666\anaconda3\envs\PVN\lib\site-packages\torch\nn_reduction.py:49: UserWarning: size_average and reduce args will be deprecated, please use reduction='mean' instead. warnings.warn(warning.format(ret)) {'bn_decay': 0.5, 'bn_momentum': 0.9, 'cal_metrics': False, 'checkpoint': None, 'cls': 'ape', 'decay_step': 200000.0, 'epochs': 1000, 'eval_net': False, 'lr': 0.01, 'lr_decay': 0.5, 'run_name': 'sem_seg_run_1', 'test': False, 'test_occ': False, 'weight_decay': 0} epochs: 0%| | 0/25 [00:00<?, ?it/s] Traceback (most recent call last): File "C:\Users\666\anaconda3\envs\PVN\lib\runpy.py", line 193, in _run_module_as_main "main", mod_spec) File "C:\Users\666\anaconda3\envs\PVN\lib\runpy.py", line 85, in _run_code exec(code, run_globals) File "D:\PVN3D-master\PVN3D-master\pvn3d\train\train_linemod_pvn3d.py", line 537, in best_loss=best_loss File "D:\PVN3D-master\PVN3D-master\pvn3d\train\train_linemod_pvn3d.py", line 363, in train for ibs, batch in enumerate(train_loader): File "C:\Users\666\anaconda3\envs\PVN\lib\site-packages\torch\utils\data\dataloader.py", line 819, in iter return _DataLoaderIter(self) File "C:\Users\666\anaconda3\envs\PVN\lib\site-packages\torch\utils\data\dataloader.py", line 560, in init w.start() File "C:\Users\666\anaconda3\envs\PVN\lib\multiprocessing\process.py", line 105, in start self._popen = self._Popen(self) File "C:\Users\666\anaconda3\envs\PVN\lib\multiprocessing\context.py", line 223, in _Popen return _default_context.get_context().Process._Popen(process_obj) File "C:\Users\666\anaconda3\envs\PVN\lib\multiprocessing\context.py", line 322, in _Popen return Popen(process_obj) File "C:\Users\666\anaconda3\envs\PVN\lib\multiprocessing\popen_spawn_win32.py", line 65, in init reduction.dump(process_obj, to_child) File "C:\Users\666\anaconda3\envs\PVN\lib\multiprocessing\reduction.py", line 60, in dump ForkingPickler(file, protocol).dump(obj) TypeError: can't pickle module objects (PVN) 666@Ali MINGW64 /d/PVN3D-master/PVN3D-master/pvn3d $ Traceback (most recent call last): File "", line 1, in File "C:\Users\666\anaconda3\envs\PVN\lib\multiprocessing\spawn.py", line 105, in spawn_main exitcode = _main(fd) File "C:\Users\666\anaconda3\envs\PVN\lib\multiprocessing\spawn.py", line 115, in _main self = reduction.pickle.load(from_parent) EOFError: Ran out of input ^C (PVN

How to solve this error? I have tried many methods but still not working and I have no idea what is the reason?! thanks in advance.