[fix]fix dropout ratio attrs error

00mjk · Feb 12, 2019 · 3ef5e15 · 3ef5e15
1 parent 79af602
commit 3ef5e15
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnx2caffe/_operators.py b/onnx2caffe/_operators.py
@@ -235,7 +235,7 @@ def _convert_dropout(node,graph,err):
     node_name = node.name
     input_name = str(node.inputs[0])
     output_name = str(node.outputs[0])
-    ratio = node.attrs.get('radio', 0.5)
+    ratio = node.attrs.get('ratio', 0.5)
     layer = myf("Dropout", node_name, [input_name], [output_name], dropout_ratio =ratio)
     graph.channel_dims[output_name] = graph.channel_dims[input_name]
     return layer