search

luopeixiang / named_entity_recognition

中文命名实体识别（包括多种模型：HMM，CRF，BiLSTM，BiLSTM+CRF的具体实现）
2.12k stars 538 forks source link

请问batch_size是在哪里设置的 #43

Open GiveMeAnApple99 opened 2 years ago

GiveMeAnApple99 commented 2 years ago

遇到了CUDA out of memory的问题，想要试着减小batch_size