Incorrect name when exporting to onnx
With some recent changes, the onnx output names are defined by each task, e.g. the ClassificationTask outputs {self.model_name}_p{class}
. The self.model_name
is set when the model is instantiated. This means that using the --name
in the onnx exort model doesn't do what we expect. This is (probably) a simple fix.