search

lmb-freiburg / flownet2

FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks
https://lmb.informatik.uni-freiburg.de/Publications/2017/IMKDB17/
Other
1k stars 318 forks source link

Multi-gpu #87

Closed youssefoumate closed 7 years ago

youssefoumate commented 7 years ago

Hello, I'm wondering if I can run "run-flownet.py" with multiple gpus ?

thank you

nikolausmayer commented 7 years ago

Hi, the script takes a --gpu parameter, so I think you should be able to just run two instances of the script in two terminals at the same time (see https://github.com/lmb-freiburg/flownet2/blob/master/scripts/run-flownet.py#L18).