| model: | |
| in_channels: 12 | |
| patch_size: 8 | |
| spectral_patch_size: 3 | |
| embed_dim: 32 | |
| encoder_depth: 1 | |
| encoder_heads: 4 | |
| decoder_dim: 32 | |
| decoder_depth: 1 | |
| decoder_heads: 4 | |
| mask_ratio: 0.90 | |
| spectral_angle_weight: 0.1 | |
| spectral_gradient_weight: 0.1 | |
| stages: | |
| - name: stage1 | |
| dataset: fMoW-Sentinel | |
| image_size: 96 | |
| train_path: ./data/stage1_train.npz | |
| train_samples: 2 | |
| epochs: 1 | |
| - name: stage2 | |
| dataset: BigEarthNet | |
| image_size: 128 | |
| train_path: ./data/stage2_train.npz | |
| train_samples: 2 | |
| epochs: 1 | |
| data: | |
| test_path: ./data/stage2_test.npz | |
| test_samples: 1 | |
| protocol: spectralgpt_progressive_s2_v2 | |
| training: | |
| batch_size: 1 | |
| learning_rate: 0.0001 | |
| weight_decay: 0.05 | |
| save_dir: ./result/checkpoints | |
| checkpoint: ./result/checkpoints/final.pth | |
| metrics: ./result/training/metrics.json | |
| amp: true | |
| runtime: | |
| device: cpu | |
| seed: 42 | |
| output_dir: ./result/output | |