Test model outputs equivalence (#6445)

* Test model outputs equivalence

* Fix failing tests

* From dict to kwargs

* DistilBERT

* Addressing @sgugger and @patrickvonplaten's comments
This commit is contained in:
Lysandre Debut
2020-08-13 11:59:35 -04:00
committed by GitHub
parent 54c687e97c
commit f7cbc13db7
4 changed files with 197 additions and 31 deletions

View File

@@ -37,6 +37,8 @@ class T5ModelTester:
self.batch_size = 13
self.encoder_seq_length = 7
self.decoder_seq_length = 9
# For common tests
self.seq_length = self.decoder_seq_length
self.is_training = True
self.use_attention_mask = True
self.use_labels = True