search

swook / EVE

Towards End-to-end Video-based Eye-tracking. ECCV 2020. https://ait.ethz.ch/eve
MIT License
115 stars 22 forks source link

train with multiple gpus #3

Open chaeyoun opened 3 years ago

chaeyoun commented 3 years ago

Did anyone train refinenet with multiple GPUs? I tried to use Dataparallel but seems like it doesn't work.

swook commented 3 years ago

Thanks for opening this issue. I myself did not previously have the chance to train EVE on multiple GPUs... so am unaware of which parts make it incompatible for nn.DataParallel. I am very much open to contributions in this direction, so anyone is very welcome to open a pull request for this purpose.

Thanks again.