对硬件要求很高？
#1
by
jikehukai
- opened
16G显存都只能用batchsize=1才能跑一下？
ubert是将每种类型都与文本拼接得到一条样本，实际的batch_size == batchsize*num_labels，你可以设置该参数 num_labels 小一点，具体参考：https://github.com/IDEA-CCNL/Fengshenbang-LM/blob/main/fengshen/examples/ubert/README.md