Closed tzuhanchang closed 6 months ago
Use a custom softmax function, modified from torch_geometric.utils.softmax. This is a temporary fix to solve unconstrained value error raised by torch.onnx.dynamo_export.
softmax
torch_geometric.utils.softmax
torch.onnx.dynamo_export
Use a custom
softmaxfunction, modified fromtorch_geometric.utils.softmax. This is a temporary fix to solve unconstrained value error raised bytorch.onnx.dynamo_export.