We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3a27ed6 commit cc4499aCopy full SHA for cc4499a
tiatoolbox/models/architecture/grandqc.py
@@ -164,6 +164,8 @@ def __init__(
164
kernel_size: int,
165
padding: int = 0,
166
stride: int = 1,
167
+ *,
168
+ bias: bool = False,
169
) -> None:
170
"""Initialize Conv2dReLU block.
171
@@ -182,6 +184,8 @@ def __init__(
182
184
Padding applied to the input. Defaults to 0.
183
185
stride (int):
186
Stride of the convolution. Defaults to 1.
187
+ bias (bool):
188
+ If `True`, adds a learnable bias to the output. Default: `False`
189
190
"""
191
norm = nn.BatchNorm2d(out_channels)
@@ -192,7 +196,7 @@ def __init__(
192
196
kernel_size,
193
197
stride=stride,
194
198
padding=padding,
195
- bias=False,
199
+ bias=bias,
200
)
201
202
activation = nn.ReLU(inplace=True)
0 commit comments