Spaces:

Kelmoir
/

FoodVisionMini

Sleeping

Kelmoir commited on Jul 8

Commit

c070ebb

verified ·

1 Parent(s): b11d3e4

Update model.py

corrected seeds setting

Files changed (1) hide show

model.py CHANGED Viewed

@@ -27,7 +27,7 @@ def create_effnetb2_model(output_classes:int=3,
   #5. Change the classification head
   #Set seed
-  set_seeds(42)
   model.classifier = nn.Sequential(
       nn.Dropout(p=0.3, inplace=True),
       nn.Linear(in_features=1408,

   #5. Change the classification head
   #Set seed
+  torch.manual_seed(42)
   model.classifier = nn.Sequential(
       nn.Dropout(p=0.3, inplace=True),
       nn.Linear(in_features=1408,