typo

michael-mueller-git · Nov 27, 2021 · 34d2856 · 34d2856
1 parent ce300b2
commit 34d2856
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -28,6 +28,6 @@ To the standard question, I can say: I can not provide an pretrained Model! I do
 
 ## Current State
 
-- Model 1 convergence when training, firs results are good
+- Model 1 convergence when training, first results are good
 - Model 2 do not convergence when training
 - Model 3 WIP
diff --git a/model/model3.py b/model/model3.py
@@ -18,13 +18,12 @@ class Model3(nn.Module):
     def __init__(self):
         super().__init__()
 
-        reduce = 1
         self.conv1 = CvT(
                 image_size=CONFIG[MODEL]['img_height'],
-                in_channels = CONFIG[MODEL]['img_channels']
+                in_channels = CONFIG[MODEL]['img_channels'],
+                dim = 16
         )
 
-        reduce += 1
         self.conv2 = CvT(
                 image_size=CONFIG[MODEL]['img_height'],
                 in_channels = 16,