Merge pull request #146 from zsef123/patch-1

Missin in_chans
2022-02-04 08:19:48 +01:00 · 2022-02-04 08:19:48 +01:00 · e9a4a1a848
parent 5d5fbbd753 6a322d8a91
commit e9a4a1a848
1 changed files with 1 additions and 1 deletions
--- a/patchconvnet_models.py
+++ b/patchconvnet_models.py
@ -194,7 +194,7 @@ class ConvStem(nn.Module):
        self.num_patches = num_patches

        self.proj = torch.nn.Sequential(
-            conv3x3(3, embed_dim // 8, 2),
+            conv3x3(in_chans, embed_dim // 8, 2),
            nn.GELU(),
            conv3x3(embed_dim // 8, embed_dim // 4, 2),
            nn.GELU(),