Run inference for summarization

allenai / longformer

Longformer: The Long-Document Transformer

https://arxiv.org/abs/2004.05150

Apache License 2.0

2.05k stars 276 forks source link

Run inference for summarization #193

Open jacob-parnell-rozetta opened 3 years ago

jacob-parnell-rozetta commented 3 years ago

Hi @ibeltagy Great work on the paper!

What parameters are specific to running the inference for summarization? I have fine-tuned the longformer-encdec-large-16384 model on a dataset, and am wanting to test the best model evaluation over the test set.

All that is saved to the output directory currently is _ckpt_epoch_1.ckpt and a folder with some additional metadata.

Apart from --test and --model_path, are there other obvious parameters that must be passed?

Thanks.

jackie930 commented 2 years ago

hi,

i got this running well for the inference part, feel free to check: https://github.com/jackie930/Noval-character-extraction/tree/main/source/longformer-model