Spaces:

SafaAsgar
/

FoodVision_mini

Sleeping

SafaAsgar commited on Oct 24, 2023

Commit

7bb9c59

•

1 Parent(s): b710963

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -14,7 +14,7 @@ def create_effnetb2_model(num_classes:int=3,
         param.requires_grad = False
-    set_seeds()
     model.classifier = nn.Sequential(
         nn.Dropout(p=0.3, inplace=True),

         param.requires_grad = False
+    torch.manual_seed(seed)
     model.classifier = nn.Sequential(
         nn.Dropout(p=0.3, inplace=True),