j96w / DenseFusion

"DenseFusion: 6D Object Pose Estimation by Iterative Dense Fusion" code repository
https://sites.google.com/view/densefusion
MIT License
1.08k stars 301 forks source link

Trained by multiple GPUs,Improve the training speed #226

Open githubueser opened 3 months ago

githubueser commented 3 months ago

Hello author, Thank you very much for uploading the source code of your work project. Can the program be trained by calling multiple GPUs at the same time to improve the training speed?If I use the parameters in the project, 500 epochs, each round is 30 minutes, then 500 rounds is too long, can I modify the parameters somewhere, can I improve the training speed?We look forward to hearing from you, thank you