Update configs

Author: sgreenbury

configs/encoder_processor_decoder.yaml

@@ -14,7 +14,7 @@ output:
 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