How to train on a single server with multiple GPUs

avinabsaha / ReIQA

Official implementation for CVPR2023 Paper "Re-IQA : Unsupervised Learning for Image Quality Assessment in the Wild"

https://arxiv.org/abs/2304.00451

MIT License

87 stars 7 forks source link

How to train on a single server with multiple GPUs #5

Closed BingfengHan closed 11 months ago

BingfengHan commented 11 months ago

Hi, thanks for sharing this beautiful work! I don't have different nodes, could you help to offer a script to train model on a single server with multiple GPUs? :)

avinabsaha commented 11 months ago

Hi @BingfengHan ,

You should be able to run the same code with --world-size 1 and --rank 0.

Scale the batch size based on GPU hardware.

avinabsaha commented 11 months ago

Closing this issue for now. Feel free to re-open to discuss further.