search

Jingkang50 / OpenOOD

Benchmarking Generalized Out-of-Distribution Detection
MIT License
858 stars 108 forks source link

Traning on multiple GPU #115

Closed YangParky closed 2 years ago

YangParky commented 2 years ago

Hi，thanks for your excellent work. When I tran the network on my gpus, I foud that it can only train on a single GPU. Do you know how to train on multi machines?

Jingkang50 commented 2 years ago

Sorry we are busy on some other stuff. Will fix soon

YangParky commented 2 years ago

Sorry we are busy on some other stuff. Will fix soon

Thanks for your attention reply, I have figured it out by setting num_gpus in train.sh and adding the 'net=net.module' at line 255 in openood/networks/utils.py .