| seed: 42 | |
| data: | |
| root: data | |
| format_version: clay_synthetic_v1_5_engineering_v1 | |
| train_samples: 4 | |
| test_samples: 2 | |
| image_size: 64 | |
| sensors: | |
| sentinel-2-l2a: | |
| channels: 10 | |
| gsd: 10.0 | |
| wavelengths_nm: [493, 560, 665, 704, 740, 783, 842, 865, 1610, 2190] | |
| rgb_indices: [2, 1, 0] | |
| landsat-c2l2-sr: | |
| channels: 6 | |
| gsd: 30.0 | |
| wavelengths_nm: [650, 560, 480, 860, 1600, 2200] | |
| rgb_indices: [0, 1, 2] | |
| sentinel-1-rtc: | |
| channels: 2 | |
| gsd: 10.0 | |
| wavelengths_nm: [3500, 4000] | |
| rgb_indices: [0, 1, 0] | |
| model: | |
| patch_size: 8 | |
| mask_ratio: 0.75 | |
| encoder_dim: 96 | |
| encoder_depth: 2 | |
| encoder_heads: 4 | |
| decoder_dim: 64 | |
| decoder_depth: 1 | |
| decoder_heads: 4 | |
| wave_dim: 32 | |
| wave_latents: 16 | |
| teacher_dim: 128 | |
| norm_pix_loss: false | |
| paper_model: | |
| patch_size: 8 | |
| mask_ratio: 0.75 | |
| encoder_dim: 1024 | |
| encoder_depth: 24 | |
| encoder_heads: 16 | |
| decoder_dim: 512 | |
| decoder_depth: 4 | |
| decoder_heads: 4 | |
| parameter_count: 632M_including_DINOv2_teacher | |
| encoder_parameter_count: 311M | |
| teacher: DINOv2_304M | |
| train: | |
| epochs: 1 | |
| batch_size: 2 | |
| learning_rate: 0.0005 | |
| weight_decay: 0.000001 | |
| reconstruction_weight: 0.95 | |
| representation_weight: 0.05 | |
| num_workers: 0 | |
| runtime: | |
| device: auto | |
| paths: | |
| checkpoint: result/checkpoints/clayfoundation.pt | |
| training_metrics: result/training/metrics.json | |
| inference_dir: result/output | |
| evaluation_dir: result/evaluation | |