allenai / allenai/naacl2021-longdoc-tutorial

Rouge very low using provided example

Open
#7 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
343
Forks
35
PR merge metrics
No merged PRs in 30d

Description

Hi, thanks for the demo script and nice tutorial. I tried to use the example code included in `summarization.py` as below. But I only get Rouge ~0.1 for testing data. Is there anything I didn't notice? Or I have to use all training data to get Rouge 43 as shown in the slide. Thanks!

```
PYTHONWARNINGS="ignore" CUDA_VISIBLE_DEVICES=6,7 python summarization.py \
--fp16 --batch_size 2 --grad_accum 1 --grad_ckpt \
--max_input_len 16384 --attention_window 1024
```

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.