How to get $GPU_ID when training the model?

baiksung / MeTAL

Official PyTorch implementation of "Meta-Learning with Task-Adaptive Loss Function for Few-Shot Learning" (ICCV2021 Oral)

62 stars 13 forks source link

Closed yewanzhuo closed 2 years ago

yewanzhuo commented 2 years ago

How to get $GPU_ID when training the model? @N1T`V~P`0A0(50 R{RJFOK

baiksung commented 2 years ago

Hi,

That's just a GPU ID number. For example, if you are using GPU with ID 0, your comand would be bash MeTAL.sh 0. Hope this helps.