请问batch_size是在哪里设置的
GiveMeAnApple99 opened this issue · comments
GiveMeAnApple99 commented
遇到了CUDA out of memory的问题,想要试着减小batch_size
中文命名实体识别(包括多种模型:HMM,CRF,BiLSTM,BiLSTM+CRF的具体实现)
GiveMeAnApple99 opened this issue · comments
遇到了CUDA out of memory的问题,想要试着减小batch_size