Skip to content

Commit 3f5b125

Browse files
committed
Correct PatchEmbeddingBlock test dropout
1 parent bfb5592 commit 3f5b125

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/networks/blocks/test_patchembedding.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,19 +122,20 @@ def test_ill_arg(self):
122122
hidden_size=128,
123123
num_heads=12,
124124
proj_type="conv",
125-
dropout_rate=5.0,
125+
dropout_rate=0.1,
126126
pos_embed_type="fourier",
127127
pos_embed_kwargs=dict(scales=[1.0, 1.0]),
128128
)
129129

130+
with self.assertRaises(ValueError):
130131
PatchEmbeddingBlock(
131132
in_channels=1,
132133
img_size=(128, 128),
133134
patch_size=(16, 16),
134135
hidden_size=128,
135136
num_heads=12,
136137
proj_type="conv",
137-
dropout_rate=5.0,
138+
dropout_rate=0.1,
138139
pos_embed_type="fourier",
139140
pos_embed_kwargs=dict(scales=[1.0, 1.0, 1.0]),
140141
)

0 commit comments

Comments
 (0)