Spaces:

benbatman
/

foodvision_big_ztm_course

Runtime error

benbatman commited on Dec 21, 2022

Commit

41e29b2

•

1 Parent(s): ed98ab9

commit

Files changed (1) hide show

model.py CHANGED Viewed

@@ -29,7 +29,7 @@ def create_effnetb2_model(num_classes:int=101,
     param.requires_grad = False
   # Change the classifier head with random seed for reproducibility
-  torch.manaul_seed(seed)
   model.classifier = nn.Sequential(
       nn.Dropout(p=0.3, inplace=True),
       nn.Linear(in_features=1408, out_features=num_classes)

     param.requires_grad = False
   # Change the classifier head with random seed for reproducibility
+  torch.manual_seed(seed)
   model.classifier = nn.Sequential(
       nn.Dropout(p=0.3, inplace=True),
       nn.Linear(in_features=1408, out_features=num_classes)