again...

app.py

@@ -19,13 +19,13 @@ class MyCarClassifier(nn.Module):
         in_ch = self.backbone.fc.in_features
         self.backbone.fc = nn.Identity()
         self.fc = nn.Sequential(
-            nn.Linear(in_ch, 512),   # fc.1
+            nn.Linear(in_ch, 1024),      # fc.1
             nn.ReLU(),
-            nn.BatchNorm1d(512),     # fc.3
-            nn.Linear(512, 256),     # fc.5
+            nn.BatchNorm1d(1024),        # fc.3
+            nn.Linear(1024, 512),        # fc.5
             nn.ReLU(),
-            nn.BatchNorm1d(256),     # fc.7
-            nn.Linear(256, num_classes) # fc.9
+            nn.BatchNorm1d(512),         # fc.7
+            nn.Linear(512, num_classes)  # fc.9
         )
 
     def forward(self, x):