models timeout

recommendation/utils/ranker_1.py

@@ -10,7 +10,7 @@ class Ranker:
         self.device = device
         self.dataset_img = dataset_img
         self.batch_size = batch_size
-        self.dataloader = torch.utils.data.DataLoader(dataset_img, batch_size=batch_size, shuffle=False, num_workers=4)
+        self.dataloader = torch.utils.data.DataLoader(dataset_img, batch_size=batch_size, shuffle=False)
 
         return