hombit commited on
Commit
441bf3a
·
verified ·
1 Parent(s): ed5931c

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +9 -5
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
- | File | Output shape | Aggregation |
120
- |------|-------------|-------------|
121
- | `astromer2_mean.onnx` | `[batch, 256]` | Masked mean pooling: `sum(z * mask_in) / sum(mask_in)` |
122
- | `astromer2_max.onnx` | `[batch, 256]` | Masked max pooling over valid timesteps |
123
- | `astromer2_full.onnx` | `[batch, 200, 256]` | Full per-timestep sequence; consumer aggregates |
 
 
 
 
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