search

chxy95 / HDRTVNet

ICCV2021 - A New Journey from SDRTV to HDRTV.
MIT License
126 stars 22 forks source link

multi-gpu training #8

Open haifangqin opened 2 years ago

haifangqin commented 2 years ago

Hi, thanks for your good work!

I would like to know how to train in multi-gpus to improve the speed of training?

Looking forward to your reply! thanks a lot in advance!

chxy95 commented 2 years ago

@haifangqin Modify gpu_ids in configuration file to use mutil-gpu. And you can try to use similar commands here to perform distributed learning.

haifangqin commented 2 years ago

Thanks for the answer! It means that modify the gpu_ids will distributed automatically？ Or do I need to modify any code of your repo?

chxy95 commented 2 years ago

@haifangqin Refer to this page for distributed learning.