transductive-vos.pytorch
transductive-vos.pytorch copied to clipboard
I use two gpus, how do I modify the parameters
I use two gpus, how do I modify the parameters?thank you
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)