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