search

yuval-alaluf / SAM

Official Implementation for "Only a Matter of Style: Age Transformation Using a Style-Based Regression Model" (SIGGRAPH 2021) https://arxiv.org/abs/2102.02754
https://yuval-alaluf.github.io/SAM/
MIT License
632 stars 151 forks source link

GPU parallel #38

Closed yyy7124 closed 2 years ago

yyy7124 commented 2 years ago

Excuese me，can you tell me how to set up multiple GPUs in parallel?

yuval-alaluf commented 2 years ago

Running with multiple GPUs can be done by using nn.DataParallel. The following issue may help you get started with adding this feature: https://github.com/eladrich/pixel2style2pixel/issues/154