SatMAE / conf /config.yaml
zhangrenchao's picture
Upload SatMAE model package
355f250 verified
Raw
History Blame Contribute Delete
1.03 kB
seed: 42
data:
root: data
protocol: fmow_rgb_temporal
mode: temporal
image_size: 32
channels: 3
frames: 3
timestamp_fields: [year_offset_2002, month_zero_based, hour]
num_classes: 10
train_samples: 32
test_samples: 8
mask_ratio: 0.75
model:
architecture: satmae_vit_large_patch16
runtime_profile: smoke
image_size: 32
patch_size: 4
in_channels: 3
frames: 3
embed_dim: 64
encoder_depth: 2
encoder_heads: 4
decoder_dim: 32
decoder_depth: 1
decoder_heads: 4
mode: temporal
spectral_groups: [[0, 1], [2]]
norm_pix_loss: false
same_mask: false
spatial_mask: false
training:
epochs: 2
batch_size: 4
accum_iter: 1
base_learning_rate: 0.001
learning_rate: null
min_learning_rate: 0.0
warmup_epochs: 1
weight_decay: 0.05
num_workers: 0
resume: null
save_every: 1
runtime:
device: auto
amp: true
paths:
checkpoint: result/checkpoints/satmae.pt
training_metrics: result/training/metrics.json
inference_dir: result/output
evaluation_dir: result/evaluation