SkySense / conf /config.yaml
zhangrenchao's picture
Update SkySense model package
2d7eead verified
Raw
History Blame Contribute Delete
772 Bytes
seed: 42
data:
root: data
protocol: skysense_multires_multitemporal_v2
train_samples: 2
test_samples: 1
hr_size: 224
s2_size: 64
s1_size: 64
num_classes: 6
hr_channels: 3
s2_channels: 10
s1_channels: 2
hr_timesteps: 1
s2_timesteps: 20
s1_timesteps: 10
model:
embed_dim: 24
hr_patch_size: 16
s2_patch_size: 8
s1_patch_size: 8
temporal_depth: 1
temporal_heads: 4
num_regions: 16
prototypes_per_region: 4
train:
epochs: 1
batch_size: 1
learning_rate: 0.001
weight_decay: 0.0001
alignment_weight: 0.05
num_workers: 0
amp: true
runtime:
device: cpu
paths:
checkpoint: result/checkpoints/skysense.pt
training_metrics: result/training/metrics.json
inference_dir: result/output
evaluation_dir: result/evaluation