Spaces:

Saint5
/

food_vision_mini

Sleeping

Peter Jaimbo commited on May 26

Commit

e703d0e

1 Parent(s): 432f4ea

set_seeds() in model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -27,7 +27,7 @@ def create_effnetb2_model(num_classes:int=3,
     param.requires_grad = False
   # Change the classifier head
-  set_seeds()
   model.classifier = nn.Sequential(
       nn.Dropout(p=0.3, inplace=True),
       nn.Linear(in_features=1408,

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