📚 Fix tutorials logs

ENSTA-U2IS-AI · Nov 5, 2023 · e4fd953 · e4fd953
1 parent 5fd0ada
commit e4fd953
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 5 deletions.
diff --git a/auto_tutorials_source/tutorial_bayesian.py b/auto_tutorials_source/tutorial_bayesian.py
@@ -90,7 +90,7 @@ def optim_lenet(model: nn.Module) -> dict:
 ):
     args = init_args(datamodule=MNISTDataModule)
 
-net_name = "bayesian-lenet-mnist"
+net_name = "logs/bayesian-lenet-mnist"
 
 # datamodule
 args.root = str(root / "data")

diff --git a/auto_tutorials_source/tutorial_der_cubic.py b/auto_tutorials_source/tutorial_der_cubic.py
@@ -89,8 +89,9 @@ def optim_regression(
     "False",
 ):
     args = init_args()
+    args.use_cv = False
 
-net_name = "der-mlp-cubic"
+net_name = "logs/der-mlp-cubic"
 
 # dataset
 train_ds = Cubic(num_samples=1000)

diff --git a/auto_tutorials_source/tutorial_evidential_classification.py b/auto_tutorials_source/tutorial_evidential_classification.py
@@ -84,7 +84,7 @@ def optim_lenet(model: nn.Module) -> dict:
 ):
     args = init_args(datamodule=MNISTDataModule)
 
-net_name = "dec-lenet-mnist"
+net_name = "logs/dec-lenet-mnist"
 
 # datamodule
 args.root = str(root / "data")

diff --git a/auto_tutorials_source/tutorial_mc_dropout.py b/auto_tutorials_source/tutorial_mc_dropout.py
@@ -78,7 +78,7 @@
 ):
     args = init_args(network=ResNet, datamodule=MNISTDataModule)
 
-net_name = "mc-dropout-lenet-mnist"
+net_name = "logs/mc-dropout-lenet-mnist"
 
 # datamodule
 args.root = str(root / "data")

diff --git a/torch_uncertainty/transforms/mixup.py b/torch_uncertainty/transforms/mixup.py
@@ -159,7 +159,7 @@ def __call__(self, x: Tensor, y: Tensor) -> tuple[Tensor, Tensor]:
 
 class RegMixup(AbstractMixup):
     """RegMixup method from Pinto et al.,
-    "RegMixup: Mixup as a Regularizer Can Surprisingly Improve Accuracy and Out Distribution Robustness" (NeurIPS 2022)
+    'RegMixup: Mixup as a Regularizer Can Surprisingly Improve Accuracy and Out Distribution Robustness' (NeurIPS 2022)
     https://arxiv.org/abs/2206.14502.
     """