I use two gpus, how do I modify the parameters

microsoft / transductive-vos.pytorch

a transductive approach for video object segmentation

155 stars 32 forks source link

I use two gpus, how do I modify the parameters #5

Open cookies1109 opened 4 years ago

cookies1109 commented 4 years ago

I use two gpus, how do I modify the parameters?thank you

mikeyEcology commented 3 years ago

I ran on two GPUs by modifying the code in main.py, line 21. I changed it from os.environ['CUDA_VISIBLE_DEVICES'] = '4,5,6,7' to os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' (or use whatever device IDs you want to use)