/pointer-generator

The pytorch implementation of Get To The Point: Summarization with Pointer-Generator Networks.

Primary LanguagePython

Pointer-Generator-Pytorch

About

The pytorch implementation of Get To The Point: Summarization with Pointer-Generator Networks. This implementation also tries to use Transformer as the encoder. The project are heavily borrowed from atulkum-pointer_summarizer and jadore801120-attention-is-all-you-need-pytorch.

Requirements

  • python==3.7.4
  • pytorch==1.4.0
  • pyrouge==0.1.3
  • tensorflow>=1.13.1

Quick start

Note:

  • There is only single example repeated across the batch in the decode mode of beam search.