Skip to content

Commit 1e00077

Browse files
committed
fix: potential typo in resnet comments
1 parent e8a5feb commit 1e00077

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

labml_nn/resnet/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def __init__(self, in_channels: int, out_channels: int, stride: int):
128128
self.bn2 = nn.BatchNorm2d(out_channels)
129129

130130
# Shortcut connection should be a projection if the stride length is not $1$
131-
# of if the number of channels change
131+
# or if the number of channels change
132132
if stride != 1 or in_channels != out_channels:
133133
# Projection $W_s x$
134134
self.shortcut = ShortcutProjection(in_channels, out_channels, stride)
@@ -210,7 +210,7 @@ def __init__(self, in_channels: int, bottleneck_channels: int, out_channels: int
210210
self.bn3 = nn.BatchNorm2d(out_channels)
211211

212212
# Shortcut connection should be a projection if the stride length is not $1$
213-
# of if the number of channels change
213+
# or if the number of channels change
214214
if stride != 1 or in_channels != out_channels:
215215
# Projection $W_s x$
216216
self.shortcut = ShortcutProjection(in_channels, out_channels, stride)

0 commit comments

Comments
 (0)