mikonvergence commited on
Commit
aa88412
·
verified ·
1 Parent(s): 7f4735c

Add model config

Browse files
Files changed (1) hide show
  1. config.json +35 -0
config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "betaearth-segformer",
3
+ "model_name": "betaearth-segformer",
4
+ "description": "BetaEarth SegFormer-B2 no FiLM (ISPRS baseline) \u2014 no timestamp needed",
5
+ "architecture": {
6
+ "encoder_name": "mit_b2",
7
+ "fusion_dim": 256,
8
+ "embed_dim": 64,
9
+ "time_embed_dim": 128,
10
+ "encoder_frozen": false,
11
+ "has_film": false,
12
+ "in_channels": {
13
+ "s2_l1c": 9,
14
+ "s2_l2a": 9,
15
+ "s1_rtc": 2,
16
+ "cop_dem": 1
17
+ }
18
+ },
19
+ "training": {
20
+ "dataset": "Major-TOM (62,489 grid cells)",
21
+ "split": "49,991 train / 6,248 val / 6,250 test",
22
+ "loss": "cosine_similarity + 0.1 * MSE",
23
+ "optimizer": "AdamW",
24
+ "epochs": 20,
25
+ "resolution": "10m (1068x1068 px tiles)"
26
+ },
27
+ "performance": {
28
+ "test_cosine_similarity": 0.88,
29
+ "test_lulc_accuracy": 0.869,
30
+ "trainable_params": "104.8M",
31
+ "total_params": "104.8M"
32
+ },
33
+ "inputs": "S2 L1C+L2A (9ch), S1 RTC (2ch), COP-DEM (1ch)",
34
+ "output": "(H, W, 64) float32, L2-normalised per pixel"
35
+ }