Commit b5ff88b0 by Siju Committed by Tianqi Chen

[Darknet] softmax temperature in frontend (#1429)

parent 3b476ffe
...@@ -260,6 +260,9 @@ def _darknet_reshape(inputs, attrs): ...@@ -260,6 +260,9 @@ def _darknet_reshape(inputs, attrs):
def _darknet_softmax_output(inputs, attrs): def _darknet_softmax_output(inputs, attrs):
"""Process the softmax operation.""" """Process the softmax operation."""
temperature = attrs.get('temperature', 1)
if temperature != 1:
inputs[0] = inputs[0] / float(temperature)
op_name, new_attrs = 'softmax', {} op_name, new_attrs = 'softmax', {}
if _darknet_parse_bool_str(attrs, 'multi_output'): if _darknet_parse_bool_str(attrs, 'multi_output'):
new_attrs['axis'] = 1 new_attrs['axis'] = 1
...@@ -529,6 +532,8 @@ def _get_darknet_attrs(net, layer_num): ...@@ -529,6 +532,8 @@ def _get_darknet_attrs(net, layer_num):
elif LAYERTYPE.SOFTMAX == layer.type: elif LAYERTYPE.SOFTMAX == layer.type:
attr.update({'axis' : 1}) attr.update({'axis' : 1})
attr.update({'use_flatten' : True}) attr.update({'use_flatten' : True})
if layer.temperature:
attr.update({'temperature' : str(layer.temperature)})
elif LAYERTYPE.SHORTCUT == layer.type: elif LAYERTYPE.SHORTCUT == layer.type:
add_layer = net.layers[layer.index] add_layer = net.layers[layer.index]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment