yjxiong / temporal-segment-networks

Code & Models for Temporal Segment Networks (TSN) in ECCV 2016
BSD 2-Clause "Simplified" License
1.54k stars 477 forks source link

Extract optical flow with gpu #181

Closed haofanwang closed 6 years ago

haofanwang commented 6 years ago

When run command bash scripts/extract_optical_flow.sh SRC_FOLDER OUT_FOLDER NUM_WORKER,is there a way to select which GPU to use?

haofanwang commented 6 years ago

nvm.Closed.

zzw1123 commented 6 years ago

Hi, could you please share the solution? And what is the meaning of "nvm"?

yjxiong commented 6 years ago

@zzw1123 https://stackoverflow.com/questions/39649102/how-do-i-select-which-gpu-to-run-a-job-on

zzw1123 commented 6 years ago

@yjxiong Thanks for the answer. I am also wondering that if the code can run on CPU in stead of GPU?

yjxiong commented 6 years ago

Yes. But it could be slow.

zzw1123 commented 6 years ago

OK! Thanks a lot for your help!