File size: 997 Bytes
e3a0c55 |
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 53 |
dataloader:
_target_: earthtext.datamodules.chipmultilabel.ChipMultilabelModule
metadata_file: /opt/data/california-worldcover-chips/california-worldcover-chips-osm-multilabels_mod-prec70pc_emb-v1s2.parquet
embeddings_folder: /opt/data/california-worldcover-chips/embeddings_v1.0
chips_folder: /opt/data/clay-california-worldcover-rgbnir-vvvh-chips/chips
get_osm_strlabels: true
get_osm_ohearea: true
get_osm_ohecount: true
get_osm_ohelength: true
embeddings_normalization: mean_stdev_norm
multilabel_threshold_osm_ohecount: 1
normalize_input: true
batch_size: 32
model:
_target_: earthtext.models.osmencoder.OSMEncoder
output_dim: 768
layers_spec:
- 512
- 1024
- 2048
use_osm_counts: true
use_osm_lengths: true
osm_tags_indexes:
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 13
- 14
- 18
- 23
- 30
- 33
- 37
- 43
- 58
- 61
- 68
- 71
- 84
- 87
- 93
- 95
- 96
- 98
activation_fn: relu
training:
loss: rmse
epochs: 1
|