DOFA / conf /config.yaml
zhangrenchao's picture
Upload DOFA model package
1d4cac8 verified
Raw
History Blame Contribute Delete
1.18 kB
seed: 42
data:
root: data
protocol: dofa_multisensor_npz_v1
image_size: 224
data_range: 1.0
train_samples_per_modality: 1
test_samples_per_modality: 1
modalities:
sentinel1:
channels: 2
wavelengths: [3.5, 4.0]
sentinel2:
channels: 9
wavelengths: [0.443, 0.490, 0.560, 0.665, 0.705, 0.740, 0.783, 0.842, 1.610]
naip:
channels: 3
wavelengths: [0.490, 0.560, 0.665]
gaofen:
channels: 4
wavelengths: [0.490, 0.560, 0.665, 0.830]
enmap:
channels: 202
wavelength_mode: synthetic_uniform
wavelength_start: 0.420
wavelength_step: 0.006
model:
image_size: 224
patch_size: 16
embed_dim: 32
encoder_depth: 1
encoder_heads: 4
decoder_dim: 24
decoder_depth: 1
decoder_heads: 4
wavelength_embed_dim: 16
hypernetwork_heads: 4
mask_ratio: 0.75
training:
epochs: 1
batch_size: 1
learning_rate: 0.0001
weight_decay: 0.05
num_workers: 0
amp: true
runtime:
device: auto
inference_batch_size: 1
paths:
checkpoint: result/checkpoints/dofa.pt
training_metrics: result/training/metrics.json
inference_dir: result/output
evaluation_dir: result/evaluation