elmo-bilstm-cnn-crf where to modify the batch size?

where to modify the batch size?

Open negacy opened this issue 5 years ago • 2 comments

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

Thanks.

Mar 16 '19 16:03 negacy

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

Mar 19 '19 16:03 nreimers

Thank you!

Mar 20 '19 03:03 negacy