Description
Model dlrm
ERR_export: Failed to convert the exported program to an ONNX model. [96mThis is step 3/3[0m of exporting the model to ONNX. Next steps:- If there is a missing ONNX function implement it and register it to the registry.- If there is an internal error during ONNX conversion debug the error and summit a PR to PyTorch.- Create an error report with torch.onnx.export(... report=True) and save the ExportedProgram as a pt2 file. Create an issue in the PyTorch GitHub repository against the [96monnx[0m component. Attach the error report and the pt2 model.## Exception summary<class 'AssertionError'>: padding_idx must not be None. This is likely a dispatcher error<class 'torch.onnx._internal.exporter._errors.GraphConstructionError'>: Error when calling function 'TracedOnnxFunction(<function aten_embedding_bag_padding_idx at 0x7f265502ff40>)' with args '[SymbolicTensor('p_model_emb_l_0_weight' type=Tensor(FLOAT16) shape=[100000064] producer=None index=None) SymbolicTensor('args_2_0' type=Tensor(INT64) shape=[100] producer=None index=None) SymbolicTensor('select' type=Tensor(INT64) shape=[1] producer=node_Gather_5 index=0) False 0 True None False None]' and kwargs '{}'<class 'torch.onnx._internal.exporter._errors.ConversionError'>: Error when translating node %embedding_bag : [num_users=1] = call_function[target=torch.ops.aten.embedding_bag.padding_idx](args = (%p_model_emb_l_0_weight %args_2_0 %select False 0 True None False None) kwargs = {}). See the stack trace for more information.(Refer to the full stack trace above for more information.)