Spaces:

jcarnero
/

birds-classification

Sleeping

jcarnero commited on Mar 13, 2023

Commit

613c01a

1 Parent(s): 92b515f

fix model classes to 200

Files changed (1) hide show

deployment/model.py CHANGED Viewed

@@ -15,7 +15,7 @@ def get_model() -> nn.Sequential:
         nn.ReLU(inplace=True),
         nn.BatchNorm1d(512),
         nn.Dropout(0.5),
-        nn.Linear(512, 37, bias=False),
     )
     return nn.Sequential(net, head)

         nn.ReLU(inplace=True),
         nn.BatchNorm1d(512),
         nn.Dropout(0.5),
+        nn.Linear(512, 200, bias=False),
     )
     return nn.Sequential(net, head)