Spaces:

sankhyikii
/

A_Dog_Or_A_Cat

Build error

sankhyikii commited on Feb 28, 2023

Commit

5564a23

1 Parent(s): 9815b9f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,8 +18,8 @@ for param in model.parameters():
 # # Replace the classifier layer
 # # to customise it according to our output
 model.classifier = nn.Sequential(
-    nn.Linear(36, 256),
-    # nn.Linear(1024, 256),
     nn.Linear(256, 2))
 checkpoint = torch.load(

 # # Replace the classifier layer
 # # to customise it according to our output
 model.classifier = nn.Sequential(
+    nn.Linear( 256*7*7, 1024),
+    nn.Linear(1024, 256),
     nn.Linear(256, 2))
 checkpoint = torch.load(