想问一下是单卡还是多卡呢，我看论文是单卡17G，我用RTX3090 24G单卡显示爆显存，是为什么呢？能解答一下吗

YaoZhang93 / mmFormer

[MICCAI 2022] The official code for "mmFormer: Multimodal Medical Transformer for Incomplete Multimodal Learning of Brain Tumor Segmentation"

Apache License 2.0

93 stars 10 forks source link

Open nnnnnnml opened 2 months ago

nnnnnnml commented 2 months ago

想问一下是单卡还是多卡呢，我看论文是单卡17G，我用RTX3090 24G单卡显示爆显存，是为什么呢？能解答一下吗

YaoZhang93 commented 2 months ago

4卡训练每块卡17G，单卡训练需要把batch size缩小4倍