luopeixiang / named_entity_recognition

中文命名实体识别(包括多种模型:HMM,CRF,BiLSTM,BiLSTM+CRF的具体实现)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

请问batch_size是在哪里设置的

GiveMeAnApple99 opened this issue · comments

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