ylabbe / cosypose

Code for "CosyPose: Consistent multi-view multi-object 6D pose estimation", ECCV 2020.
MIT License
301 stars 89 forks source link

Configuration with runjob-config #48

Closed yupei-git closed 3 years ago

yupei-git commented 3 years ago

Hello, when I got to the 'Training details' in the 'BOP20 models and results' section, using 'runjob --ngpus=1 python -m cosypose.scripts.run_detector_training --config bop-ycbv-synt+real' to train the ycbv dataset on the synthetic + real image, the problem occursScreenshot from 2021-04-26 18-10-10

I modified the job-runner-config.yaml, as shows:

Screenshot from 2021-04-26 18-11-15 and used 'runjob-config job-runner-config.yaml', but got the same error.

Before that, the 'Installation' and 'Downloading and preparing data' sections were successfully completed. Since I wanted to do it with a single local GPU, I just used the command line 'export CUDA_VISIBLE_DEVICES=0' and 'python -m cosypose.scripts.example_multigpu' in the 'Note on GPU parallization' section. But that works in 'Reproducing single-view results' part.

Could anyone help me with this? Thanks in advance!

yupei-git commented 3 years ago

That is solved. :)

JohannesAma commented 3 years ago

Hello @yupei-git , can you tell me how you solved this ?

yupei-git commented 3 years ago

I reinstalled cosypose and the environment. Then copied the local_data from the original cosypose into the new one. When I ran the train again, the above error did not exist.