We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8dcec2b commit cadd0faCopy full SHA for cadd0fa
hls4ml/converters/pytorch/core.py
@@ -65,7 +65,7 @@ def parse_batchnorm_layer(pytorch_layer, layer_name, input_shapes, data_reader,
65
66
# batchnorm para
67
layer['epsilon'] = pytorch_layer.eps
68
- layer['use_gamma'] = layer['use_beta'] = not pytorch_layer.affine
+ layer['use_gamma'] = layer['use_beta'] = pytorch_layer.affine
69
70
in_size = 1
71
for dim in input_shapes[0][1:]:
0 commit comments