farrosalferro24 commited on
Commit
8aa4a9a
1 Parent(s): 30c1de8

Update script.py

Browse files
Files changed (1) hide show
  1. script.py +1 -1
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=True)
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"))