Update config.yaml

interTwin-eu · Oct 14, 2024 · 949f84c · 949f84c
1 parent 967e244
commit 949f84c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/use-cases/mnist/torch/config.yaml b/use-cases/mnist/torch/config.yaml
@@ -94,6 +94,7 @@ inference_pipeline:
             num_workers_dataloader: 4
             pin_gpu_memory: true
             shuffle_test: false
+          strategy: ${strategy}
 
       - class_path: saver.TorchMNISTLabelSaver
         init_args: