🐛 Fix missing MNIST parameter

ENSTA-U2IS-AI · Jun 30, 2023 · 1843fa9 · 1843fa9
1 parent dd63306
commit 1843fa9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torch_uncertainty/datamodules/mnist.py b/torch_uncertainty/datamodules/mnist.py
@@ -186,4 +186,7 @@ def add_argparse_args(
         p.add_argument("--batch_size", type=int, default=128)
         p.add_argument("--val_split", type=int, default=0)
         p.add_argument("--num_workers", type=int, default=4)
+        p.add_argument(
+            "--evaluate_ood", dest="ood_detection", action="store_true"
+        )
         return parent_parser