Spaces:

aashu-0
/

Food101_pytorch

Sleeping

aashu-0 commited on Jan 31

Commit

7444c8b

verified ·

1 Parent(s): af26cfe

update

Files changed (1) hide show

model.py CHANGED Viewed

@@ -4,8 +4,8 @@ import torchvision
 from torch import nn
-create_effnetb2_model(num_classes: int=101,
-                        seed: int =42):
   #weights, transforms and model instance
   weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT

 from torch import nn
+def create_effnetb2_model(num_classes: int,
+                          seed: int =42):
   #weights, transforms and model instance
   weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT