Export fixes/improvements
Export fixes/improvements:
- Correctly assign onnx outputs
- Use model name for probability outputs in onnx and predictionwriter
- Save more metadata to onnx file
- Normalisation happens inside the onnx model, not in athena (that appears to be unsupported)
Minor things:
- remove
ruamel.yaml
- use
torch.set_float32_matmul_precision("medium")
- vertex task bugfix
- onnx export: default to
r22default
track selection, use model name to build outputs, rename output to"network.onnx"
- add some
torch.manual_seed()
Edited by Samuel Van Stroud