Spaces:
Sleeping
Sleeping
Update model.py
Browse files
model.py
CHANGED
@@ -14,7 +14,7 @@ def create_effnetb2_model(num_classes:int=3,
|
|
14 |
param.requires_grad = False
|
15 |
|
16 |
|
17 |
-
|
18 |
|
19 |
model.classifier = nn.Sequential(
|
20 |
nn.Dropout(p=0.3, inplace=True),
|
|
|
14 |
param.requires_grad = False
|
15 |
|
16 |
|
17 |
+
torch.manual_seed(seed)
|
18 |
|
19 |
model.classifier = nn.Sequential(
|
20 |
nn.Dropout(p=0.3, inplace=True),
|