Spaces:

Thenujan
/

Covid_CNN

Runtime error

Thenujan commited on Oct 5, 2023

Commit

7cd7988

1 Parent(s): ba9fd81

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -34,7 +34,7 @@ class ConvModel(nn.Module):
         self.classifier = nn.Sequential(
                                 nn.Flatten(),
-                                nn.Linear(in_features = 40960, out_features = output_shape))
     def forward(self, x):

         self.classifier = nn.Sequential(
                                 nn.Flatten(),
+                                nn.Linear(in_features = 2560, out_features = output_shape))
     def forward(self, x):