Upload README.md with huggingface_hub
Browse files
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.
|
|
|
|
| 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.
|
| 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
|