Spaces:

gulnuravci
/

spongebob_character_identifier

Running

gulnuravci commited on Nov 24, 2023

Commit

fced91e

1 Parent(s): b59b6e9

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -38,7 +38,7 @@ def create_effnetb0_model(num_classes:int=10,
   torch.manual_seed(seed)
   model.classifier = nn.Sequential(
       nn.Dropout(p=0.3),
-      nn.Linear(in_features=1280], out_features=num_classes)
   )
   return model, transforms

   torch.manual_seed(seed)
   model.classifier = nn.Sequential(
       nn.Dropout(p=0.3),
+      nn.Linear(in_features=1280, out_features=num_classes)
   )
   return model, transforms