Merge pull request #106 from alan-turing-institute/fix-default-config-name

Fix default config name (#81)

Author: sgreenbury

configs/encoder_processor_decoder.yaml

@@ -6,15 +6,15 @@ defaults:
   - _self_
 
 seed: 42
-experiment_name: processor
+experiment_name: encoder_processor_decoder
 output:
   checkpoint_name: encoder_processor_decoder.ckpt
   save_config: true
 
 training:
   n_steps_input: 1
   n_steps_output: 4
-  stride: null
+  stride: 4
   autoencoder_checkpoint: null
   freeze_autoencoder: false
 

configs/model/encoder_processor_decoder.yaml

@@ -4,8 +4,8 @@ defaults:
   - /processor: flow_matching
   - _self_
 
-learning_rate: 0.001
-train_processor_only: false
+learning_rate: 0.0001
+train_processor_only: true
 teacher_forcing_ratio: 0.5
 max_rollout_steps: 10
 loss_func:

configs/processor/flow_matching.yaml

@@ -11,7 +11,7 @@ backbone:
   in_channels: null
   out_channels: null
   cond_channels: null
-  mod_features: 256
+  mod_features: 200
   hid_channels: [32, 64, 128]
   hid_blocks: [2, 2, 2]
   spatial: 2

src/autocast/eval/encoder_processor_decoder.py

@@ -72,8 +72,8 @@ def parse_args() -> argparse.Namespace:
     )
     parser.add_argument(
         "--config-name",
-        default="processor",
-        help="Hydra config name to compose (defaults to 'processor').",
+        default="encoder_processor_decoder",
+        help="Hydra config name to compose (defaults to 'encoder_processor_decoder').",
     )
     parser.add_argument(
         "overrides",
@@ -455,8 +455,8 @@ def main() -> None:
     resolved_cfg = OmegaConf.to_container(cfg, resolve=True)
     wandb_logger, _ = create_wandb_logger(
         cfg.get("logging"),
-        experiment_name=cfg.get("experiment_name", "processor"),
-        job_type="evaluate-processor",
+        experiment_name=cfg.get("experiment_name", "encoder_processor_decoder"),
+        job_type="evaluate-encoder-processor-decoder",
         work_dir=work_dir,
         config={
             "hydra": resolved_cfg,

src/autocast/train/encoder_processor_decoder.py

@@ -51,8 +51,8 @@ def parse_args() -> argparse.Namespace:
     )
     parser.add_argument(
         "--config-name",
-        default="processor",
-        help="Hydra config name to compose (defaults to 'processor').",
+        default="encoder_processor_decoder",
+        help="Hydra config name to compose (defaults to 'encoder_processor_decoder').",
     )
     parser.add_argument(
         "overrides",
@@ -185,8 +185,8 @@ def main() -> None:  # noqa: PLR0915
     model_cfg = cfg.get("model") or cfg
     wandb_logger, watch_cfg = create_wandb_logger(
         cfg.get("logging"),
-        experiment_name=cfg.get("experiment_name", "processor"),
-        job_type="train-processor",
+        experiment_name=cfg.get("experiment_name", "encoder_processor_decoder"),
+        job_type="train-encoder-processor-decoder",
         work_dir=work_dir,
         config={"hydra": resolved_cfg} if resolved_cfg is not None else None,
     )