search

kentaroy47 / BlazeFace_Person.pytorch

Unofficial Pytorch implementation of :fire:BlazeFace:fire:
MIT License
42 stars 10 forks source link

multi-GPU #3

Closed Cherry2410 closed 4 years ago

Cherry2410 commented 4 years ago

Hi, how to set multi-GPU to train?

kentaroy47 commented 4 years ago

https://pytorch.org/tutorials/beginner/blitz/data_parallel_tutorial.html Try wrapping it with data parallel!