search

tangjiapeng / SA-ConvONet

ICCV2021 Oral SA-ConvONet: Sign-Agnostic Optimization of Convolutional Occupancy Networks
MIT License
89 stars 5 forks source link

How can I call multiple GPUs during training #4

Open miaowang525 opened 2 years ago

miaowang525 commented 2 years ago

Hello @tangjiapeng Your work is great and i am very interested in your article, and I have some questions to ask you. I have five GPUs, how to call multiple GPUs to run code during training?

tangjiapeng commented 2 years ago

Hi, thanks a lot for your interest.

The way to call the multiple GPUs to run the test-time optimization is similar to what we usually do in the multi-GPUs training (i.e. using the torch.nn.DataParallel module).

And if you use multi-GPUs, you can increase the batch size, then you can accelerate the test-time optimization.