Use dynamic batch dimensions for `ONNX` input samples

tzuhanchang / HyPER

Hypergraph for Particle Event Reconstruction

MIT License

3 stars 0 forks source link

Use dynamic batch dimensions for `ONNX` input samples #56

Closed tzuhanchang closed 1 month ago

tzuhanchang commented 1 month ago

The new ONNX model is shown below, with dynamic dimensions s0 (number of nodes), s1 (number of edges), s3 (number of global features) and s5 (number of hyperedges).

HyPER onnx

Close #55 after merging.