search

er-muyue / DeFRCN

MIT License
181 stars 43 forks source link

Help with multi-GPU training in Google Colab #61

Closed suryasid09 closed 1 year ago

suryasid09 commented 1 year ago

Hello,

I am trying to execute the run_voc.sh file in the Google Colab. I have extracted the 'Base Pre Train' part from the run_voc.sh file and running it directly from the notebook. I haven't changed anything in the code and following the step-by-step procedure as mentioned in the DeFRCN github website. I get the following error.

mulitgpu

But, when I use --num-gpus 1, the code seems to work. I want to increase the speed of training as currently is very slow. I would really appreciate help in this regard.

suryasid09 commented 1 year ago

I got it, the Google Colab only provide 1 GPU.