File size: 1,034 Bytes
355f250 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | 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
|