Skip to content

GitLab

Unverified Commit c1a13861 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix ViTMAEModelTester (#17470) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent b0e0ac8a
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
tests/models/vit_mae/test_modeling_tf_vit_mae.py
View file @ c1a13861
......@@ -107,6 +107,10 @@ class TFViTMAEModelTester:
num_hidden_layers=self.num_hidden_layers,
num_attention_heads=self.num_attention_heads,
intermediate_size=self.intermediate_size,
decoder_hidden_size=self.hidden_size,
decoder_num_hidden_layers=self.num_hidden_layers,
decoder_num_attention_heads=self.num_attention_heads,
decoder_intermediate_size=self.intermediate_size,
hidden_act=self.hidden_act,
hidden_dropout_prob=self.hidden_dropout_prob,
attention_probs_dropout_prob=self.attention_probs_dropout_prob,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment