isaaccorley commited on
Commit
cea52ae
·
verified ·
1 Parent(s): 96f5b35

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -11,14 +11,15 @@ with no imagery or labels at inference; the embedding is Matryoshka-structured,
11
  deploy width.
12
 
13
  ## Files
14
- - `mind.pt` - trained ReSIREN encoder (embed_dim 3072, depth 12). Load with `mind_standalone.load_mind`.
 
15
  - `mind.onnx` (+ `mind.onnx.data`) - ONNX graph for onnxruntime / TensorRT (input `latlon` [N,2], dynamic batch).
16
  - `mind.pt2` - torch ExportedProgram; load with `torch.export.load`.
17
 
18
  ## Usage
19
  ```python
20
  from mind_standalone import load_mind, embed # github.com/taylor-geospatial/mind
21
- emb = embed(load_mind("mind.pt"), lats, lons) # [N, 64]; half=True for ~12x on GPU
22
  ```
23
  Build a TensorRT engine from the ONNX with `trtexec --onnx=mind.onnx --saveEngine=mind.trt --fp16`.
24
  Code and paper: https://github.com/taylor-geospatial/mind
 
11
  deploy width.
12
 
13
  ## Files
14
+ - `mind.safetensors` - recommended weights, fp16 (227 MB, safe / no-pickle). Load with `mind_standalone.load_mind`.
15
+ - `mind.pt` - fp32 weights (454 MB) for exact reproduction.
16
  - `mind.onnx` (+ `mind.onnx.data`) - ONNX graph for onnxruntime / TensorRT (input `latlon` [N,2], dynamic batch).
17
  - `mind.pt2` - torch ExportedProgram; load with `torch.export.load`.
18
 
19
  ## Usage
20
  ```python
21
  from mind_standalone import load_mind, embed # github.com/taylor-geospatial/mind
22
+ emb = embed(load_mind("mind.safetensors"), lats, lons) # [N, 64]; half=True for ~12x on GPU
23
  ```
24
  Build a TensorRT engine from the ONNX with `trtexec --onnx=mind.onnx --saveEngine=mind.trt --fp16`.
25
  Code and paper: https://github.com/taylor-geospatial/mind