PawelGlomski-Intel opened a new pull request #20555: URL: https://github.com/apache/incubator-mxnet/pull/20555
## Description ## sg_mkldnn_selfatt_qk and sg_mkldnn_selfatt_valatt ops, **when not quantized**, are not exporting their attributes to the \*-symbol.json file, thus making the SymbolBlock.imports/model.load_checkpoint functions crash with an error: "mxnet.base.MXNetError: MXNetError: Required parameter heads of int is not presented, in operator sg_mkldnn_selfatt_\*(name="sg_mkldnn_selfatt_\*")" when trying to load a model that uses them. This change fixes this issue. ## Checklist ## ### Essentials ### - [ ] PR's title starts with a category (e.g. [BUGFIX], [MODEL], [TUTORIAL], [FEATURE], [DOC], etc) - [ ] Changes are complete (i.e. I finished coding on this PR) - [ ] All changes have test coverage - [ ] Code is well-documented ### Changes ### - [ ] Assign attributes of transformer operators -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
