optimum
8a00085a - Create ONNX configurations with loss as output (#142)

Commit

3 years ago

Create ONNX configurations with loss as output (#142) * Create wrapper classes OnnxConfigWithLoss, OnnxConfigWithPastAndLoss, nnxSeq2SeqConfigWithPastAndLoss * Add tests for the wrappers

References

#142 - Create ONNX configurations with loss as output

Author

JingyaHuang

Parents

dfe330a2

optimum 8a00085a - Create ONNX configurations with loss as output (#142)

optimum
8a00085a - Create ONNX configurations with loss as output (#142)