Skip to content
This repository has been archived by the owner on Nov 16, 2023. It is now read-only.

Daden/bartt5 add text summarization for BART and T5 models #591

Open
wants to merge 14 commits into
base: staging
Choose a base branch
from

Conversation

daden-ms
Copy link
Contributor

@daden-ms daden-ms commented May 26, 2020

Description

added bart/t5 finetuning and test functionality
added a notebook example for t5-small model with performance verified

Related Issues

Checklist:

  • My code follows the code style of this project, as detailed in our contribution guidelines.
  • I have added tests.
  • I have updated the documentation accordingly.

@daden-ms daden-ms requested a review from saidbleik May 26, 2020 20:33
@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

Review Jupyter notebook visual diffs & provide feedback on notebooks.


Powered by ReviewNB

@daden-ms daden-ms changed the title Daden/bartt5 add text stigmatization for BART and T5 models Daden/bartt5 add text summarization for BART and T5 models May 26, 2020
@daden-ms
Copy link
Contributor Author

for bart-large model, finetuned 10000 steps and ROUGE score for cnn/dm test dataset is
image

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant