risenyard commited on
Commit
ae6b313
路
verified 路
1 Parent(s): d5099bc

Add files using upload-large-folder tool

Browse files
Files changed (2) hide show
  1. README.md +17 -19
  2. data/encoder/checkpoint/encoder.pt +3 -0
README.md CHANGED
@@ -10,8 +10,8 @@ tags:
10
 
11
  # EGMS-QA 路 Encoder
12
 
13
- Frozen spatio-temporal encoder and its 10,000-tile token cache for **EGMS-QA**,
14
- a natural-language question-answering system over
15
  [European Ground Motion Service](https://egms.land.copernicus.eu/) (EGMS)
16
  persistent-scatterer displacement time series.
17
 
@@ -25,28 +25,26 @@ Companion repos: [`risenyard/egms-qa-dataset`](https://huggingface.co/datasets/r
25
 
26
  ## Files
27
 
28
- | file | what it is |
29
  |---|---|
30
- | `encoder.pt` | frozen encoder checkpoint (`model` state dict + training args) |
31
- | `encoder_tokens_10k.pt` | token cache for the 10k tiles: `spatial_tokens [10000,65,256]` (X), `token_mask [10000,65]` (m, empty cells = 0), plus `tile_ids`, `splits`, `point_count_per_bin`, `n_points_per_tile`, `metadata` |
32
- | `encoder_tokens_10k_metadata.json` | extraction metadata |
33
- | `manifest/split.parquet` | 10k-tile train/val/test split manifest |
34
- | `manifest/data_config.json` | data configuration reference |
 
35
 
36
  ## Use
37
 
38
- ```python
39
- from huggingface_hub import hf_hub_download
40
- import torch
41
 
42
- tok = torch.load(hf_hub_download("risenyard/egms-qa-encoder", "encoder_tokens_10k.pt"),
43
- map_location="cpu", weights_only=False)
44
- X, m = tok["spatial_tokens"], tok["token_mask"] # [10000,65,256], [10000,65]
 
 
45
  ```
46
 
47
- Re-extracting tokens from raw tiles, or loading the checkpoint architecture,
48
- needs the model code in the GitHub repo (`egms_encoder`). Place these files at
49
- `data/encoder/鈥 of a checkout to reproduce end to end (see the repo README).
50
-
51
- The raw EGMS Level-3 product is Copernicus data and is not redistributed here.
52
  Licensed CC-BY-4.0.
 
10
 
11
  # EGMS-QA 路 Encoder
12
 
13
+ Frozen spatio-temporal encoder checkpoint for **EGMS-QA**, a natural-language
14
+ question-answering system over
15
  [European Ground Motion Service](https://egms.land.copernicus.eu/) (EGMS)
16
  persistent-scatterer displacement time series.
17
 
 
25
 
26
  ## Files
27
 
28
+ | path | what it is |
29
  |---|---|
30
+ | `data/encoder/checkpoint/encoder.pt` | frozen encoder checkpoint (`model` state dict + training args) |
31
+
32
+ The precomputed **token cache** and the **raw tiles** live in the dataset repo
33
+ [`risenyard/egms-qa-dataset`](https://huggingface.co/datasets/risenyard/egms-qa-dataset)
34
+ (`data/encoder/tokens/`, `data/tiles/`); the split manifest and normalization
35
+ are in the code repo under `data/encoder/`.
36
 
37
  ## Use
38
 
39
+ Load the checkpoint with the model code in the GitHub repo (`egms_encoder`), then
40
+ re-extract tokens from the raw tiles:
 
41
 
42
+ ```bash
43
+ python -m egms_encoder.extract_tokens \
44
+ --checkpoint data/encoder/checkpoint/encoder.pt \
45
+ --manifest data/encoder/manifest/split.parquet \
46
+ --output-dir outputs/tokens
47
  ```
48
 
49
+ Most users can skip this and use the released token cache from the dataset repo.
 
 
 
 
50
  Licensed CC-BY-4.0.
data/encoder/checkpoint/encoder.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cdb0bd2437c063a79c871eb54445dbee6fb82bd9e20186768fa5b947ffd9d474
3
+ size 79442395