EMSANet icon indicating copy to clipboard operation
EMSANet copied to clipboard

Published 20 hours ago • verified publisher icon TUI-NICR

Train using multi-gpu

Open yhs-ai opened this issue 1 year ago • 1 comments

Dear authors,

Would it be possible to train the model using multi-gpus?

yhs-ai avatar Apr 15 '23 13:04 yhs-ai

Currently, we do not support mulit-gpu training. Luckily, we have access to A100 GPUs with 40GB/80GB, which have enough VRAM for training with batch size 8.

danielS91 avatar Apr 20 '23 14:04 danielS91