fix incorrect channel size for first shortcut

2024-05-18 02:28:35 +01:00
parent cb3fd47324
commit 9f6852ee23
1 changed files with 1 additions and 1 deletions
--- a/resnet.py
+++ b/resnet.py
@@ -99,7 +99,7 @@ class MaiRes(nn.Module):
                in_channels=64,
                out_channels=128,
                stride=2,
-                shortcut=projection_shortcut(in_channels=64, out_channels=64),
+                shortcut=projection_shortcut(in_channels=64, out_channels=128),
            ),
            ResidualBlock(
                in_channels=128,