mboudiaf/RePRI-for-Few-Shot-Segmentation

How to set GPU?

AmingWu opened this issue · 0 comments

Dear author:

When I run this command:

bash scripts/train.sh pascal 0 [5] 50

I want to run the code on the fifth GPU. There is an error about "cuda out of memory".
However, this GPU has 12 GiB free.