kentonl / e2e-coref

End-to-end Neural Coreference Resolution
Apache License 2.0
524 stars 173 forks source link

does this support batch_size > 1 ? #29

Closed kalpitdixit closed 6 years ago

kentonl commented 6 years ago

Unfortunately, no. The long documents already provide natural parallelism, and the memory is quite constrained as is. The implementation of the EMNLP 2017 version in AllenNLP supports batching, but it doesn't have all the bells and whistles required for the best accuracy. If trading off accuracy for speed is what you need, I recommend checking them out (https://github.com/allenai/allennlp/blob/master/allennlp/models/coreference_resolution/coref.py)