Slim config.yaml to inference-only TimeSeriesModel architecture

config.yaml

@@ -1,67 +1,3 @@
-train_data_path: ./forge_synth_data
-model_path: ./forge_checkpoints
-model_name: Forge_1M
-continue_training: false
-checkpoint_path: null
-seed: 2025
-wandb: false
-wandb_project_name: Forge
-wandb_plots: false
-batch_size: 400
-num_training_iterations: 150000
-validation_batch_size: 64
-num_validation_batches: 1
-num_workers: 16
-prefetch_arrow_batches: true
-dataloader_prefetch_factor: 4
-gradient_accumulation_enabled: false
-accumulation_steps: 1
-log_interval: 1024
-save_every: 25000
-memory_debug_interval: 0
-memory_debug_tracemalloc: false
-memory_debug_growth_mb: 40
-memory_debug_include_children: true
-generator_proportions:
-  forecast_pfn: 1.0
-  gp: 1.0
-  kernel: 1.0
-  sawtooth: 1.0
-  sinewave: 1.0
-  step: 1.0
-  anomaly: 1.0
-  spike: 1.0
-  cauker_univariate: 1.0
-  ou_process: 3.0
-  augmented_per_sample_2048: 2.0
-  augmented_temp_batch_2048: 2.0
-lr_scheduler: cosine
-peak_lr: 0.0005
-min_lr_ratio: 0.01
-warmup_ratio: 0.003
-stable_ratio: 0.9
-decay_type: cosine
-num_cycles: 0.5
-num_restart_cycles: 4
-weight_decay: 0.01
-beta1: 0.9
-beta2: 0.98
-optimizer_eps: 1e-6
-gradient_clip_val: 100.0
-scaler: custom_robust
-gift_eval:
-  evaluate_on_gift_eval: false
-  max_context_length: 3072
-  create_plots: false
-  max_plots: 5
-  dataset_storage_path: null
-data_augmentation:
-  nan_augmentation: true
-  scaler_augmentation: false
-  length_shortening: true
-  nan_stats_path: ../tempopfn/src/data/nan_stats.json
-augmentation_probabilities:
-  scaler_augmentation: 0.5
 TimeSeriesModel:
   embed_size: 128
   num_encoder_layers: 3