6 Commits

Author	SHA1	Message	Date
Sam Shleifer	e9e6efdc45	BartForSequenceClassification: fix num_labels, add test (#3110)	2020-03-03 15:54:29 -05:00
Sam Shleifer	b54ef78d0c	Bart-CNN (#3059) ... `generate` code that produces 99% identical summarizations to fairseq on CNN test data, with caching.	2020-03-02 10:35:53 -05:00
Julien Chaumond	f5516805c2	Fix bart slow test	2020-02-26 20:47:49 +00:00
Julien Chaumond	9cda3620b6	Fix (non-slow) tests on GPU (torch) (#3024) ... * Fix tests on GPU (torch) * Fix bart slow tests Co-authored-by: Sam Shleifer <sshleifer@gmail.com>	2020-02-26 11:59:25 -05:00
Sam Shleifer	92487a1dc0	Bart: fix layerdrop and cached decoder_input_ids for generation (#2969)	2020-02-22 16:25:04 -05:00
Sam Shleifer	53ce3854a1	New BartModel (#2745) ... * Results same as fairseq * Wrote a ton of tests * Struggled with api signatures * added some docs	2020-02-20 18:11:13 -05:00