training rel classification/detection using multi gpus

yuweihao / KERN

Code for Knowledge-Embedded Routing Network for Scene Graph Generation (CVPR 2019)

MIT License

120 stars 35 forks source link

training rel classification/detection using multi gpus #8

Open MitraTj opened 5 years ago

MitraTj commented 5 years ago

Hi,

Is there any solution to use more than one gpu for training relation classification/detection?

Thanks!

yuweihao commented 5 years ago

Hi @MitraTj , I suppose the code could be adjusted to use multi gpus but it is sorry that I have no time to develop it. One of disadvantages of the model of this repo is time consuming. This repo is based on the wonderful repo neural-motifs and their models are trained much faster (but also only one gpu for training relation classificaiton/detection).

MitraTj commented 5 years ago

Fine. Thanks!