farrosalferro24
commited on
Commit
•
8aa4a9a
1
Parent(s):
30c1de8
Update script.py
Browse files
script.py
CHANGED
@@ -30,7 +30,7 @@ class PytorchWorker:
|
|
30 |
|
31 |
model = timm.create_model(model_name,
|
32 |
num_classes=number_of_categories,
|
33 |
-
pretrained=
|
34 |
|
35 |
# if not torch.cuda.is_available():
|
36 |
# model_ckpt = torch.load(model_path, map_location=torch.device("cpu"))
|
|
|
30 |
|
31 |
model = timm.create_model(model_name,
|
32 |
num_classes=number_of_categories,
|
33 |
+
pretrained=False)
|
34 |
|
35 |
# if not torch.cuda.is_available():
|
36 |
# model_ckpt = torch.load(model_path, map_location=torch.device("cpu"))
|