Updating config

earthlab · Jun 29, 2023 · c67ffb9 · c67ffb9
1 parent 57378b7
commit c67ffb9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/config.yaml b/config.yaml
@@ -3,4 +3,4 @@ aws:
 torch:
   inference:
     batch_size: 100
-    num_workers: 12
+    num_workers: 5
diff --git a/src/ml/inference.py b/src/ml/inference.py
@@ -102,7 +102,7 @@ def inference_torch(model_file: TrainedModel, tile_match_file: Union[SingleRegio
             output = model(data)
             probs = torch.softmax(output, dim=1)
             conf, pred = torch.max(probs, 1)
-
+            print(data.shape)
             print(pred)
 
             # store res to file