How to change parameters when num_device =2

Hzzone / ProPos

Self-Supervised Learning for Deep Clustering (TPAMI 2022)

104 stars 15 forks source link

How to change parameters when num_device =2 #14

Closed congxin0920 closed 1 year ago

congxin0920 commented 1 year ago

First of all, thank you very much for your code, since there are only two Gpus, I would like to ask you if there are any parameters that need to be changed？

Hzzone commented 1 year ago

Closed as I have not conducted any experiments on 2 GPUs. I am so sorry. I may suggest SimSiam, which only costs 1 GPU. https://github.com/Hzzone/ProPos/issues/13