FTAG: Adding support in Athena for Gn1 Aux tasks
Can support multiple output types for the onnx models -
floatvec<float>vec<char>
Introducing GNNConfig::Config
- The goal is to eventually get rid of
lwtnndependency. - Right now, we look at
GNNConfig::Configfor creating the decorators. (Will include configs for accessors as well in future)
- Decorators are created from reading the outputs of the onnx model and not from the metadata as before
- (In future, it can be extended for the accessors as well)
Changes to the model output format
- Previous models were returning
pb,pc,puscores asvec<float>(of length 1). We had to extract the first element and then decorate it as afloat - The new models return the exact type it needs to be decorated as. so
floatforpb,vec<char>fortrack_classetc. - Everything is figured out from the onnx model directly, and it doesn't require to read the metadata for it anymore.
Backward compatibility is assured!
Edited by Nilotpal Kakati