🐛 Fix experiment scripts

ENSTA-U2IS-AI · Dec 26, 2023 · ea3d66a · ea3d66a
1 parent e212876
commit ea3d66a
Show file tree

Hide file tree

Showing 7 changed files with 9 additions and 8 deletions.
diff --git a/experiments/classification/cifar10/wideresnet.py b/experiments/classification/cifar10/wideresnet.py
@@ -33,4 +33,4 @@
         **vars(args),
     )
 
-    cli_main(model, dm, root, args.exp_dir, args.exp_name, args)
+    cli_main(model, dm, args.exp_dir, args.exp_name, args)
diff --git a/experiments/classification/cifar100/resnet.py b/experiments/classification/cifar100/resnet.py
@@ -33,4 +33,4 @@
         **vars(args),
     )
 
-    cli_main(model, dm, root, args.exp_dir, args.exp_name, args)
+    cli_main(model, dm, args.exp_dir, args.exp_name, args)
diff --git a/experiments/classification/cifar100/vgg.py b/experiments/classification/cifar100/vgg.py
@@ -33,4 +33,4 @@
         **vars(args),
     )
 
-    cli_main(model, dm, root, args.exp_dir, args.exp_name, args)
+    cli_main(model, dm, args.exp_dir, args.exp_name, args)
diff --git a/experiments/classification/cifar100/wideresnet.py b/experiments/classification/cifar100/wideresnet.py
@@ -33,4 +33,4 @@
         **vars(args),
     )
 
-    cli_main(model, dm, root, args.exp_dir, args.exp_name, args)
+    cli_main(model, dm, args.exp_dir, args.exp_name, args)
diff --git a/experiments/classification/mnist/bayesian_lenet.py b/experiments/classification/mnist/bayesian_lenet.py
@@ -59,4 +59,4 @@ def optim_lenet(model: nn.Module) -> dict:
         **vars(args),
     )
 
-    cli_main(baseline, dm, root, "logs/", net_name, args)
+    cli_main(baseline, dm, "logs/", net_name, args)
diff --git a/experiments/classification/mnist/lenet.py b/experiments/classification/mnist/lenet.py
@@ -30,7 +30,8 @@ def optim_lenet(model: nn.Module) -> dict:
     else:
         root = Path(args.root)
 
-    net_name = "std-lenet-mnist"
+    if args.exp_name == "":
+        args.exp_name = "std-lenet-mnist"
 
     # datamodule
     args.root = str(root / "data")
@@ -48,4 +49,4 @@ def optim_lenet(model: nn.Module) -> dict:
         **vars(args),
     )
 
-    cli_main(baseline, dm, root, "logs/", net_name, args)
+    cli_main(baseline, dm, args.exp_dir, args.exp_name, args)
diff --git a/experiments/classification/tiny-imagenet/resnet.py b/experiments/classification/tiny-imagenet/resnet.py
@@ -30,7 +30,7 @@ def optim_tiny(model: nn.Module) -> dict:
 
     # net_name = f"{args.version}-resnet{args.arch}-tiny-imagenet"
     if args.exp_name == "":
-        args.exp_name = f"{args.version}-resnet{args.arch}-cifar10"
+        args.exp_name = f"{args.version}-resnet{args.arch}-tinyimagenet"
 
     # datamodule
     args.root = str(root / "data")