How can I train the other backbone at LUPerson use 2 or 4 gpus?

DengpanFu / LUPerson

Unsupervised Pre-training for Person Re-identification (LUPerson)

242 stars 35 forks source link

How can I train the other backbone at LUPerson use 2 or 4 gpus? #24

Open puyiwen opened 11 months ago

puyiwen commented 11 months ago

Hi, it is a great work! I want to train the other backbone like mbv2. I dont know how to change the trainning set with 2 or 4 GPUs, like --lr or --bs. Can you help me? Thank you very much!!