File size: 908 Bytes
387a20d 7a2d30b 387a20d 7a2d30b 387a20d 7a2d30b 387a20d 7a2d30b 387a20d 7a2d30b 387a20d 7a2d30b 387a20d | 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 | model:
in_channels: 12
patch_size: 8
spectral_patch_size: 3
embed_dim: 32
encoder_depth: 1
encoder_heads: 4
decoder_dim: 32
decoder_depth: 1
decoder_heads: 4
mask_ratio: 0.90
spectral_angle_weight: 0.1
spectral_gradient_weight: 0.1
stages:
- name: stage1
dataset: fMoW-Sentinel
image_size: 96
train_path: ./data/stage1_train.npz
train_samples: 2
epochs: 1
- name: stage2
dataset: BigEarthNet
image_size: 128
train_path: ./data/stage2_train.npz
train_samples: 2
epochs: 1
data:
test_path: ./data/stage2_test.npz
test_samples: 1
protocol: spectralgpt_progressive_s2_v2
training:
batch_size: 1
learning_rate: 0.0001
weight_decay: 0.05
save_dir: ./result/checkpoints
checkpoint: ./result/checkpoints/final.pth
metrics: ./result/training/metrics.json
amp: true
runtime:
device: cpu
seed: 42
output_dir: ./result/output
|