| #Base directories - change these to switch datasets | |
| base_data_dir: "/Volumes/T9/Data_FOXES" # Change this line for different datasets | |
| base_checkpoint_dir: "/Volumes/T9/Data_FOXES" # Change this line for different datasets | |
| wavelengths: [94, 131, 171, 193, 211, 304, 335] # AIA wavelengths in Angstroms | |
| # GPU configuration | |
| # Options: | |
| # - Single GPU: gpu_ids: 0 or gpu_ids: [0] | |
| # - Multi GPU: gpu_ids: [0, 1] (uses both GPU 0 and 1) | |
| # - All GPUs: gpu_ids: "all" (uses all available GPUs) | |
| # - CPU only: gpu_ids: -1 | |
| gpu_ids: "all" # Use both GPUs | |
| # Model configuration | |
| selected_model: "ViTLocal" | |
| batch_size: 6 | |
| epochs: 150 | |
| oversample: false | |
| balance_strategy: "" | |
| calculate_base_weights: false # Whether to calculate class-based weights for loss function | |
| vit_architecture: | |
| embed_dim: 256 | |
| num_channels: 7 | |
| num_classes: 1 | |
| patch_size: 8 | |
| num_patches: 4096 | |
| hidden_dim: 1024 | |
| num_heads: 8 | |
| num_layers: 8 | |
| dropout: 0.1 | |
| learning_rate: 0.0001 | |
| # Data paths (automatically constructed from base directories) | |
| data: | |
| aia_dir: | |
| "${base_data_dir}/AIA_processed" | |
| sxr_dir: | |
| "${base_data_dir}/SXR_processed" | |
| sxr_norm_path: | |
| "${base_data_dir}/SXR_processed/normalized_sxr.npy" | |
| checkpoints_dir: | |
| "${base_checkpoint_dir}/new-checkpoint/" | |
| wandb: | |
| entity: "" # Set to your W&B username or team name | |
| project: FOXES | |
| job_type: training | |
| tags: | |
| - aia | |
| - sxr | |
| - regression | |
| run_name: run_1 | |
| notes: AIA to SXR translation |