search

xingyizhou / CenterNet2

Two-stage CenterNet
Apache License 2.0
1.19k stars 189 forks source link

Training GPU Setup #47

Open bakerada opened 3 years ago

bakerada commented 3 years ago

📚 Documentation

I have seen it noted the GPUs used for inference. I was hoping you could share what GPU setup you used to train the models, particularly the R2-101-DCN-BiFPN variants.

xingyizhou commented 3 years ago

Hi, Sorry for the delayed response. I used 8 24G GPUs (Quadro RTX 6000) to train the model. If you don't have large memory GPUs, decreasing the batchsize and learning rate and increasing the total iterations accordingly should also be fine.

nsankar90 commented 2 years ago

Thanks for the great work Xingyizhou

It could be great if you include on how to setup worker for distributed learning!