Skip to content

Commit cc4499a

Browse files
committed
🎨 Improve structure of the code.
1 parent 3a27ed6 commit cc4499a

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tiatoolbox/models/architecture/grandqc.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,8 @@ def __init__(
164164
kernel_size: int,
165165
padding: int = 0,
166166
stride: int = 1,
167+
*,
168+
bias: bool = False,
167169
) -> None:
168170
"""Initialize Conv2dReLU block.
169171
@@ -182,6 +184,8 @@ def __init__(
182184
Padding applied to the input. Defaults to 0.
183185
stride (int):
184186
Stride of the convolution. Defaults to 1.
187+
bias (bool):
188+
If `True`, adds a learnable bias to the output. Default: `False`
185189
186190
"""
187191
norm = nn.BatchNorm2d(out_channels)
@@ -192,7 +196,7 @@ def __init__(
192196
kernel_size,
193197
stride=stride,
194198
padding=padding,
195-
bias=False,
199+
bias=bias,
196200
)
197201

198202
activation = nn.ReLU(inplace=True)

0 commit comments

Comments
 (0)