Update utils.py

utils.py

@@ -244,7 +244,7 @@ class MolecularPropertyPredictionModel():
         # construct the dataloader
         test_loader = torch.utils.data.DataLoader(
             test_dataset,
-            batch_size=16,
+            batch_size=32,
             collate_fn=self.data_collator,
         )
 
@@ -266,13 +266,13 @@ class MolecularPropertyPredictionModel():
 
         return y_pred
 
-    @spaces.GPU(duration=20)
+    @spaces.GPU(duration=30)
     def predict_long(self, valid_df, task_type):
         test_dataset = Dataset.from_pandas(valid_df)
         # construct the dataloader
         test_loader = torch.utils.data.DataLoader(
             test_dataset,
-            batch_size=16,
+            batch_size=32,
             collate_fn=self.data_collator,
         )