oneformer3d icon indicating copy to clipboard operation
oneformer3d copied to clipboard

About multi-GPU training learning rate

Open RayYoh opened this issue 1 year ago • 1 comments

Hi @filaPro, recently, I have been trying to reproduce the results of Oneformer3D to compare with my own method. Since using just one GPU would be relatively slow, I changed to using multi-GPU training to align the configs with my method.
My question is if I use 3 or 4 GPUs for training, which means the batch size will be 34 or 44, do I need to modify the learning rate in the raw configs (i.e. 310-4 or 410-4).

Best

RayYoh avatar Sep 20 '24 01:09 RayYoh

We never tried multi-gpu for oneforner3d, so probably yes, multiplying should be fine, but i'm not sure if after it the metrics will be reproduced exactly.

filaPro avatar Sep 20 '24 08:09 filaPro