Spaces:

Th3BossC
/

FoodVision

Runtime error

Th3BossC commited on May 5, 2023

Commit

8388c79

1 Parent(s): 7f5fb3e

pls be done with errors

Files changed (1) hide show

model.py CHANGED Viewed

@@ -20,7 +20,7 @@ def create_model():
         # nn.Linear(in_features = 1024, out_features = 512, bias = True),
         # nn.ReLU(),
         # nn.Dropout(p = 0.3, inplace = True),
-        nn.Linear(in_features = 1024, out_features = num_classes, bias = True)
     )

         # nn.Linear(in_features = 1024, out_features = 512, bias = True),
         # nn.ReLU(),
         # nn.Dropout(p = 0.3, inplace = True),
+        nn.Linear(in_features = 1024, out_features = 101, bias = True)
     )