Commit Graph
9 Commits
Author SHA1 Message Date
Julien Chaumond 1d18930462 Harmonize no_cuda flag with other scripts 2019-12-09 20:37:55 -05:00
Rémi Louf f7eba09007 clean for release 2019-12-09 20:37:55 -05:00
Rémi Louf 2a64107e44 improve device usage 2019-12-09 20:37:55 -05:00
Rémi Louf c0707a85d2 add README 2019-12-09 20:37:55 -05:00
Rémi Louf ade3cdf5ad integrate ROUGE 2019-12-09 20:37:55 -05:00
Rémi Louf 076602bdc4 prevent BERT weights from being downloaded twice 2019-12-09 20:37:55 -05:00
Rémi Louf a1994a71ee simplified model and configuration 2019-12-09 20:37:55 -05:00
Rémi Louf 3a9a9f7861 default output dir to documents dir 2019-12-09 20:37:55 -05:00
Rémi Louf 2403a66598 give transformers API to BertAbs 2019-12-09 20:37:55 -05:00