Fix TFRemBertEncoder all_hidden_states (#15510)

* fix

* fix test

* remove expected_num_hidden_layers

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-02-04 17:32:14 +01:00
committed by GitHub
parent 854a0d526c
commit bbe9c6981b
2 changed files with 1 additions and 4 deletions

View File

@@ -90,9 +90,6 @@ class TFRemBertModelTester:
self.num_choices = 4
self.scope = None
# RemBERT also returns the upprojected word embeddings as an hidden layers
self.expected_num_hidden_layers = self.num_hidden_layers + 2
def prepare_config_and_inputs(self):
input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size)