Merge pull request #41 from ENSTA-U2IS/add-mimo

✨ Add MIMO
ENSTA-U2IS-AI · Aug 25, 2023 · a0f1348 · a0f1348
2 parents 368c1e9 + f8829c6
commit a0f1348
Show file tree

Hide file tree

Showing 20 changed files with 936 additions and 72 deletions.
diff --git a/README.md b/README.md
@@ -52,6 +52,7 @@ To date, the following deep learning baselines have been implemented:
 - Deep Ensembles
 - BatchEnsemble
 - Masksembles
+- MIMO
 - Packed-Ensembles (see [blog post](https://medium.com/@adrien.lafage/make-your-neural-networks-more-reliable-with-packed-ensembles-7ad0b737a873))
 - Bayesian Neural Networks
 

diff --git a/experiments/classification/cifar10/wideresnet.py b/experiments/classification/cifar10/wideresnet.py
@@ -14,7 +14,7 @@
 
     args = init_args(WideResNet, CIFAR10DataModule)
 
-    net_name = f"{args.version}-wideresnet{args.arch}-cifar10"
+    net_name = f"{args.version}-wideresnet28x10-cifar10"
 
     # datamodule
     args.root = str(root / "data")
@@ -26,7 +26,7 @@
         in_channels=dm.num_channels,
         loss=nn.CrossEntropyLoss,
         optimization_procedure=get_procedure(
-            f"resnet{args.arch}", "cifar10", args.version
+            "wideresnet28x10", "cifar10", args.version
         ),
         style="cifar",
         **vars(args),