开发者社区 问答 正文

为什么Bert Large模型在Nvidia V100 16G显卡上的batch size通常较小?

为什么Bert Large模型在Nvidia V100 16G显卡上的batch size通常较小?

展开
收起
萝卜丝丸子 2024-08-05 21:46:52 152 分享 版权
1 条回答
写回答
取消 提交回答
  • Bert Large模型在Nvidia V100 16G显卡上的batch size通常较小(如2-8),主要是因为该模型对显存消耗较大。batch size的具体值还会受到Embedding大小、Sequence Length等因素的影响。image.png

    2024-08-05 22:33:47
    赞同 展开评论