ONNX
brunosan commited on
Commit
e3a0c55
·
verified ·
1 Parent(s): 841a645

Model v1 and earth2text v1

Browse files
Clay-1.0.5.7_epoch-13_val-loss-0.3098.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e41ece5741b41a4e4fdb57199e353702ec34524e334017b543ebdaad3d90184
3
+ size 1726997970
Clay-text2earth-v1.0.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1d538f87e88b827b36cd8590c8a357fd38ac9738918b97dfbe8cadbd3194da4a
3
+ size 16952268
Clay-text2earth-v1.0.yaml ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataloader:
2
+ _target_: earthtext.datamodules.chipmultilabel.ChipMultilabelModule
3
+ metadata_file: /opt/data/california-worldcover-chips/california-worldcover-chips-osm-multilabels_mod-prec70pc_emb-v1s2.parquet
4
+ embeddings_folder: /opt/data/california-worldcover-chips/embeddings_v1.0
5
+ chips_folder: /opt/data/clay-california-worldcover-rgbnir-vvvh-chips/chips
6
+ get_osm_strlabels: true
7
+ get_osm_ohearea: true
8
+ get_osm_ohecount: true
9
+ get_osm_ohelength: true
10
+ embeddings_normalization: mean_stdev_norm
11
+ multilabel_threshold_osm_ohecount: 1
12
+ normalize_input: true
13
+ batch_size: 32
14
+ model:
15
+ _target_: earthtext.models.osmencoder.OSMEncoder
16
+ output_dim: 768
17
+ layers_spec:
18
+ - 512
19
+ - 1024
20
+ - 2048
21
+ use_osm_counts: true
22
+ use_osm_lengths: true
23
+ osm_tags_indexes:
24
+ - 2
25
+ - 3
26
+ - 4
27
+ - 5
28
+ - 6
29
+ - 7
30
+ - 8
31
+ - 13
32
+ - 14
33
+ - 18
34
+ - 23
35
+ - 30
36
+ - 33
37
+ - 37
38
+ - 43
39
+ - 58
40
+ - 61
41
+ - 68
42
+ - 71
43
+ - 84
44
+ - 87
45
+ - 93
46
+ - 95
47
+ - 96
48
+ - 98
49
+ activation_fn: relu
50
+ training:
51
+ loss: rmse
52
+ epochs: 1