From 9f6852ee2307f66f3b0064c3b360fb4257c5b276 Mon Sep 17 00:00:00 2001
From: Kenneth <kennethnym@outlook.com>
Date: Sat, 18 May 2024 02:28:35 +0100
Subject: [PATCH] fix incorrect channel size for first shortcut

---
 resnet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/resnet.py b/resnet.py
index a1421a6..0405114 100644
--- 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,