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:
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user