[Convolution] Error while importing onnx model with weights. (#345)

* [Convolution] Error while imported onnx model has weights.

Fix the use_bias based on the input parameters to use bias (Ex: sqeezenet). Ref #336

* Review corrections.
This commit is contained in:
Siva 2018-01-30 12:51:36 +05:30 коммит произвёл Tianqi Chen
Родитель ffcb2a5ef0
Коммит eb9442e0f2
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -95,7 +95,7 @@ def _conv():
'dilations': ('dilation', (0, 0)),
'pads': ('padding', (0, 0), _revert_caffe2_pad),
'group': ('groups', 1)},
extras={'use_bias': False},
extras={'use_bias': len(inputs) == 3},
custom_check=_dimension_constraint())(inputs, attr)
return _impl
@ -113,7 +113,7 @@ def _conv_transpose():
'dilations': ('dilation', (0, 0)),
'pads': ('padding', (0, 0), _revert_caffe2_pad)},
disables=['output_shape'],
extras={'use_bias': False},
extras={'use_bias': len(inputs) == 3},
custom_check=_dimension_constraint())(inputs, attr)
return _impl