I have trained this on single gpu with cuda 11.2 and pytorch 1.8 but still confuse about multi-gpu training

[Alex Park]

how can i use multi-gpu pretrained model on single gpu. this code seems not work.
btw, i have sent mail to yutao about license of this great work ([e-mail]) expect to get a reply :)

[Tao Yu]

how can i use multi-gpu pretrained model on single gpu. this code seems not work.
btw, i have sent mail to yutao about license of this great work ([e-mail]) expect to get a reply :)

Hi, pls modify https://github.com/geekyutao/RN/blob/master/main.py#L237 from if opt.gpus > 1: to if opt.gpus >= 1.