| 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 | |