请问用的什么型号的gpu 我用的3080 运行的时候显示显存不够将batchsize调整至4才能运行但是会导致运行效率变慢所以想请教一下

wzzzd / lm_ner

基于Pytorch的命名实体识别框架，支持LSTM+CRF、Bert+CRF、RoBerta+CRF等框架

75 stars 18 forks source link

Open 20214021016 opened 2 years ago

20214021016 commented 2 years ago

或者在代码层面听该怎么修改呢？

wzzzd commented 2 years ago

你跑的是bert+crf?

khazic commented 1 year ago

你是不是把gpu拿给其他的再用 3080的显存不会那么小的