LogSoftMax export: Check temperature

apache · Dec 18, 2018 · 8e81c64 · 8e81c64
1 parent 16199ff
commit 8e81c64
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',