AlphaEarthFoundations / conf /config.yaml
zhangrenchao's picture
Add engineering reproduction package
3549cf5 verified
Raw
History Blame Contribute Delete
2.1 kB
seed: 42
data:
root: data
format_version: aef_synthetic_paper_spec_v3
train_samples: 2
test_samples: 1
image_size: 128
input_sources:
sentinel2: {timesteps: 65, channels: 5, geometry_dim: 0}
sentinel1: {timesteps: 17, channels: 5, geometry_dim: 2}
landsat: {timesteps: 21, channels: 7, geometry_dim: 0}
target_sources:
sentinel2: {channels: 5, type: continuous, weight: 1.0, loss_grid_m: 10, shift_pixels: 2, geometry_dim: 0}
sentinel1: {channels: 5, type: continuous, weight: 1.0, loss_grid_m: 10, shift_pixels: 2, geometry_dim: 2}
landsat: {channels: 7, type: continuous, weight: 1.0, loss_grid_m: 30, geometry_dim: 0}
palsar2: {channels: 3, type: continuous, weight: 1.0, loss_grid_m: 30, geometry_dim: 2}
era5_land: {channels: 12, type: continuous, weight: 1.0, loss_grid_m: 10, geometry_dim: 0}
gedi: {channels: 101, type: continuous, weight: 1.0, loss_grid_m: 20, geometry_dim: 0}
grace: {channels: 1, type: continuous, weight: 0.5, loss_grid_m: 1280, geometry_dim: 0}
glo30: {channels: 1, type: continuous, weight: 1.0, loss_grid_m: 30, geometry_dim: 0, derived_channels: undisclosed}
nlcd: {channels: 16, type: categorical, weight: 0.5, loss_grid_m: 30, geometry_dim: 0}
model:
precision_dim: 8
time_dim: 16
space_dim: 32
embedding_dim: 64
num_blocks: 1
num_heads: 4
decoder_hidden_dim: 32
vmf_kappa: 8000.0
max_geometry_dim: 2
paper_model:
precision_dim: 128
time_dim: 512
space_dim: 1024
embedding_dim: 64
num_blocks: 15
parameter_count: approximately_480M
train:
epochs: 1
batch_size: 2
learning_rate: 0.0005
weight_decay: 0.000001
reconstruction_weight: 1.0
uniformity_weight: 0.05
consistency_weight: 0.02
text_weight: 0.001
source_dropout: {sentinel2: 0.0, sentinel1: 0.3, landsat: 0.3}
frame_dropout: {sentinel2: 0.5, sentinel1: 0.3, landsat: 0.3}
num_workers: 0
runtime:
device: auto
paths:
checkpoint: result/checkpoints/alphaearthfoundations.pt
training_metrics: result/training/metrics.json
inference_dir: result/output
evaluation_dir: result/evaluation