search

UKPLab / elmo-bilstm-cnn-crf

BiLSTM-CNN-CRF architecture for sequence tagging using ELMo representations.
Apache License 2.0
388 stars 81 forks source link

where to modify the batch size? #16

Open negacy opened 5 years ago

negacy commented 5 years ago

I am trying to run this on GPU but having memory issue. Where can I modify the batch size?

Thanks.

nreimers commented 5 years ago

You can change the config dict and set a value for the entry: miniBatchSize

Have a look here with all the possible config parameters: https://github.com/UKPLab/emnlp2017-bilstm-cnn-crf/#lstm-hyperparameters

negacy commented 5 years ago

Thank you!