How do you perform training with multi-GPUs?

microsoft / Semantics-Aligned-Representation-Learning-for-Person-Re-identification

This is an implementation of AAAI'20 paper "Semantics-Aligned Representation Learning for Person Re-identification". We leverages dense semantics to address both the spatial misalignment and semantics misalignment challenges in person re-identification.

MIT License

23 stars 13 forks source link

How do you perform training with multi-GPUs? #4

Open wencoast opened 4 years ago

wencoast commented 4 years ago

How do you set up the training arguments to perform a mode training l on a cluster with multi-GPUs?

I think that it might be related to the following arguments. Could you give me more detailed suggestions?

Screenshot from 2020-07-15 16-24-06

jinx-USTC commented 4 years ago

CUDA_VISIBLE_DEVICES=0,1,2,3....multiple gpus