MRBagherifar
/

Sample-Submission

Model card Files Files and versions Community

MRBagherifar commited on Apr 28

Commit

6f255c0

•

1 Parent(s): ed97b9f

Update script.py

Files changed (1) hide show

script.py +1 -1

script.py CHANGED Viewed

@@ -38,7 +38,7 @@ class PytorchWorker:
         self.model = _load_model(model_name, model_path)
-        self.transforms = T.Compose([T.Resize((299, 299)),
                                      T.ToTensor(),
                                      T.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])

         self.model = _load_model(model_name, model_path)
+        self.transforms = T.Compose([T.Resize((380, 380)),
                                      T.ToTensor(),
                                      T.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])