Implement PreSumm

[srpkdyy]

#1006

[srpkdyy]

ExtモデルとAbsモデル両方提供されていますが、Abs側の公式プログラムにバグを確認

Traceback (most recent call last):
File "D:\job\PreSumm\src\train.py", line 140, in
test_text_abs(args)
File "D:\job\PreSumm\src\train_abstractive.py", line 333, in test_text_abs
predictor.translate(test_iter, -1)
File "D:\job\PreSumm\src\models\predictor.py", line 151, in translate
batch_data = self.translate_batch(batch)
File "D:\job\PreSumm\src\models\predictor.py", line 214, in translate_batch
return self._fast_translate_batch(
File "D:\job\PreSumm\src\models\predictor.py", line 338, in _fast_translate_batch
[alive_seq.index_select(0, select_indices),
RuntimeError: index_select(): Expected dtype int32 or int64 for index

[srpkdyy]

Indexにスコアのような少数が入ってきているようです

########## Alive Seq #############
torch.Size([5, 1])
<class 'torch.Tensor'>
tensor([[1],
[1],
[1],
[1],
[1]])
########## select_indices #############
torch.Size([5])
<class 'torch.Tensor'>
tensor([0.1432, 0.0654, 0.2367, 0.0339, 0.5090])

公式Issueでも報告されているようです
nlpyang/PreSumm#234

[kyakuno]

モデルアップロード済み。
https://storage.googleapis.com/ailia-models/presumm/cnndm-bertext.onnx.prototxt