Aux task outputs missing
I believe that there is an issue with the ONNX export of the auxiliary tasks. As far as I can tell, the following checks never evaluate to true with the current config schema:
if "track_origin" in config["data"]["labels"]:
output_names.append("track_class")
dynamic_axes["track_class"] = {0: "n_tracks"}
if "track_vertexing" in config["data"]["labels"]:
output_names.append("vertex_indices")
dynamic_axes["vertex_indices"] = {0: "n_tracks"}
Looking at the config that is generated in salt/tests/test_pipeline.py::test_GN2
I see
labels:
jet:
- flavour_label
track:
- ftagTruthOriginLabel
- ftagTruthVertexIndex
with no reference to "track_origin" or "track_vertexing". If I try to use the model that I exported with --include-aux
in Athena, I get
AthAlgSeq ERROR SG::ExcBadAuxVar: Attempt to retrieve nonexistent aux data item `::vertex_indices' (2555).
Tagging @npond
Edited by Jackson Carl Burzynski