LogSoftMax export: Check temperature

apache · Dec 18, 2018 · a2072ee · a2072ee
1 parent b34cf35
commit a2072ee
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/mxnet/contrib/onnx/mx2onnx/_op_translations.py b/python/mxnet/contrib/onnx/mx2onnx/_op_translations.py
@@ -1665,7 +1665,10 @@ def convert_logsoftmax(node, **kwargs):
  name, input_nodes, attrs = get_inputs(node, kwargs)
 
  # Converting to int
- axis = int(attrs.get("axis", 1))
+ axis = int(attrs.get("axis", -1))
+ temp = attrs.get("temperature", 'None')
+ if temp != 'None':
+ raise AttributeError("ONNX supports only temperature=None")
 
  node = onnx.helper.make_node(
  'LogSoftmax',