apmoore1 / target-extraction

Target based extraction
https://apmoore1.github.io/target-extraction/
Apache License 2.0
4 stars 1 forks source link

4 dimensional targets #6

Closed apmoore1 closed 5 years ago

apmoore1 commented 5 years ago

The Targets when they first get sent to the majority of models are of shape (Batch Size, Number Targets, Target Sequence Length), this is not compatible with the ELMO/BERT type encoder as they expect to have shape (Batch Size, Sequence Length) tensor as input.