Fix quantized localization experiment

ekut-es · Feb 13, 2024 · 6eda317 · 6eda317
1 parent ab4cda0
commit 6eda317
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/experiments/rhode_island/config.yaml b/experiments/rhode_island/config.yaml
@@ -20,7 +20,7 @@ defaults:
     - base_config
     - override dataset: ri_capsule   # Dataset configuration name
     - override features: identity        # Feature extractor configuration name (use identity for vision datasets)
-    - override model: timm_mobilenetv3_small_075      # Neural network name (for now timm_resnet50 or timm_efficientnet_lite1)
+#    - override model: timm_mobilenetv3_small_075      # Neural network name (for now timm_resnet50 or timm_efficientnet_lite1)
     - override scheduler: 1cycle         # learning rate scheduler config name
     - override augmentation: ri_augment
     - override optimizer: sgd          # Optimizer config name

diff --git a/experiments/rhode_island/experiment/quantized_localization.yaml b/experiments/rhode_island/experiment/quantized_localization.yaml
@@ -1,5 +1,8 @@
+# @package _global_
+
+
 defaults:
-  - /model: localization_net
+  - override /model: localization_net
 
 experiment_id: quantized_localization