search

FLAIR-THU / CreamFL

[ICLR 2023] Multimodal Federated Learning via Contrastive Representation Ensemble
https://arxiv.org/abs/2302.08888
84 stars 12 forks source link

System Requirements #3

Closed momin-ahmad-khan closed 11 months ago

momin-ahmad-khan commented 1 year ago

Hi, could you tell me what are the system requirements for this code. Number of gpus, and memory of each gpu. Also, what was the average training time for the experiments. I am trying to run this on a 12GB gpu but it runs out of memory. Thanks.

yqy2001 commented 1 year ago

We use one A100-40G GPU with each run takes about one day.

12G memory is not enough, but you can try to reduce the number or dim of public data and conduct experiments in a your memory-affordable setting.

Thank you.

momin-ahmad-khan commented 11 months ago

Thank you!