CI with num_hidden_layers=2 🚀🚀🚀 (#25266)
* CI with layers=2 --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -53,7 +53,7 @@ class DPTModelTester:
|
||||
is_training=True,
|
||||
use_labels=True,
|
||||
hidden_size=32,
|
||||
num_hidden_layers=4,
|
||||
num_hidden_layers=2,
|
||||
backbone_out_indices=[0, 1, 2, 3],
|
||||
num_attention_heads=4,
|
||||
intermediate_size=37,
|
||||
@@ -62,7 +62,7 @@ class DPTModelTester:
|
||||
attention_probs_dropout_prob=0.1,
|
||||
initializer_range=0.02,
|
||||
num_labels=3,
|
||||
neck_hidden_sizes=[16, 16, 32, 32],
|
||||
neck_hidden_sizes=[16, 32],
|
||||
is_hybrid=False,
|
||||
scope=None,
|
||||
):
|
||||
|
||||
Reference in New Issue
Block a user