训练Cross-Encoder的时候为什么24G的显存还不够用？

[LensGrinder4E00]

是我哪里弄错了吗？还是说就是要这么大的显存？

[lihaoyang-ruc]

在使用schema_item_classifier.py 训练cross-encoder的时候，可以尝试减少batch_size，并增大gradient_descent_step以避免显卡OOM的问题。