| seed: 42 |
| data: |
| root: data |
| format_version: prithvi_eo_2_synthetic_engineering_v1 |
| train_samples: 4 |
| test_samples: 2 |
| channels: 6 |
| frames: 4 |
| image_size: 224 |
| band_names: [Blue, Green, Red, Narrow_NIR, SWIR1, SWIR2] |
| band_codes: [B02, B03, B04, B8A, B11, B12] |
| mean: [1087.0, 1342.0, 1433.0, 2734.0, 1958.0, 1363.0] |
| std: [2248.0, 2179.0, 2178.0, 1850.0, 1242.0, 1049.0] |
| model: |
| input_size: [4, 224, 224] |
| patch_size: [1, 16, 16] |
| channels: 6 |
| mask_ratio: 0.75 |
| encoder_dim: 96 |
| encoder_depth: 2 |
| encoder_heads: 4 |
| decoder_dim: 64 |
| decoder_depth: 1 |
| decoder_heads: 4 |
| mlp_ratio: 4 |
| metadata_dropout: 0.1 |
| norm_pix_loss: false |
| paper_model_300M_TL: |
| input_size: [4, 224, 224] |
| patch_size: [1, 16, 16] |
| encoder_dim: 1024 |
| encoder_depth: 24 |
| encoder_heads: 16 |
| decoder_dim: 512 |
| decoder_depth: 8 |
| decoder_heads: 16 |
| parameter_count: 300M |
| paper_model_600M_TL: |
| input_size: [4, 224, 224] |
| patch_size: [1, 14, 14] |
| encoder_dim: 1280 |
| encoder_depth: 32 |
| encoder_heads: 16 |
| decoder_dim: 512 |
| decoder_depth: 8 |
| decoder_heads: 16 |
| parameter_count: 600M |
| paper_training: |
| samples: 4.2M |
| validation_samples: 46k |
| epochs: 400 |
| global_batch_size: 3840 |
| maximum_learning_rate: 0.0005 |
| warmup_epochs: 40 |
| weight_decay: 0.05 |
| train: |
| epochs: 1 |
| batch_size: 2 |
| learning_rate: 0.0005 |
| weight_decay: 0.000001 |
| num_workers: 0 |
| runtime: |
| device: auto |
| paths: |
| checkpoint: result/checkpoints/prithvi_eo.pt |
| training_metrics: result/training/metrics.json |
| inference_dir: result/output |
| evaluation_dir: result/evaluation |
|
|
