Antreas commited on
Commit
3f6cd84
·
verified ·
1 Parent(s): 051d7a6

Add ONNX export with parity report

Browse files
Files changed (1) hide show
  1. onnx/parity_report.json +20 -0
onnx/parity_report.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": "axiotic/ogma-micro",
3
+ "onnx_path": "onnx/model.onnx",
4
+ "opset": 17,
5
+ "inputs": {
6
+ "token_ids": "int64[batch, sequence]",
7
+ "attention_mask": "int64[batch, sequence]",
8
+ "task_token_ids": "int64[batch] (4=QRY, 5=DOC, 6=SYM)"
9
+ },
10
+ "output": "embeddings float32[batch, d_output] L2-normalized",
11
+ "parity": {
12
+ "atol": 0.0001,
13
+ "rtol": 0.0001,
14
+ "max_abs_error": 3.3527612686157227e-07,
15
+ "mean_abs_error": 6.890632420208931e-08,
16
+ "mean_cosine": 1.0,
17
+ "passed": true
18
+ },
19
+ "file_size_bytes": 9600530
20
+ }