Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -116,11 +116,15 @@ The ONNX wrapper inverts `mask_in` internally before passing it to the encoder,
|
|
| 116 |
|
| 117 |
## Outputs (ONNX)
|
| 118 |
|
| 119 |
-
|
| 120 |
-
|
| 121 |
-
|
|
| 122 |
-
|
|
| 123 |
-
| `
|
|
|
|
|
|
|
|
|
|
|
|
|
| 124 |
|
| 125 |
ONNX opset: 13.
|
| 126 |
|
|
|
|
| 116 |
|
| 117 |
## Outputs (ONNX)
|
| 118 |
|
| 119 |
+
Single file `astromer2.onnx` with three named outputs:
|
| 120 |
+
|
| 121 |
+
| Output name | Shape | Aggregation |
|
| 122 |
+
|-------------|-------|-------------|
|
| 123 |
+
| `mean` | `[batch, 256]` | Masked mean pooling: `sum(z * mask_in) / sum(mask_in)` |
|
| 124 |
+
| `max` | `[batch, 256]` | Masked max pooling over valid timesteps |
|
| 125 |
+
| `sequence` | `[batch, 200, 256]` | Per-timestep features |
|
| 126 |
+
|
| 127 |
+
Request only the output(s) you need via `session.run(["mean"], feed)` — onnxruntime will prune unused computation.
|
| 128 |
|
| 129 |
ONNX opset: 13.
|
| 130 |
|