[Tests] Add Common Test for Training + Fix a couple of bugs (#8415)
* add training tests * correct longformer * fix docs * fix some tests * fix some more train tests * remove ipdb * fix multiple edge case model training * fix funnel and prophetnet * clean gpt models * undo renaming of albert
This commit is contained in:
committed by
GitHub
parent
52040517b8
commit
9c83b96e62
@@ -42,7 +42,7 @@ class TransfoXLModelTester:
|
||||
self.mem_len = 30
|
||||
self.key_length = self.seq_length + self.mem_len
|
||||
self.clamp_len = 15
|
||||
self.is_training = True
|
||||
self.is_training = False
|
||||
self.use_labels = True
|
||||
self.vocab_size = 99
|
||||
self.cutoffs = [10, 50, 80]
|
||||
|
||||
Reference in New Issue
Block a user