search

TUI-NICR / EMSANet

EMSANet: Efficient Multi-Task RGB-D Scene Analysis for Indoor Environments
Apache License 2.0
56 stars 3 forks source link

Train using multi-gpu #8

Open yhs-ai opened 1 year ago

yhs-ai commented 1 year ago

Dear authors,

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

danielS91 commented 1 year ago

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.