Skip to content

Commit cadd0fa

Browse files
authored
Update core.py
1 parent 8dcec2b commit cadd0fa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

hls4ml/converters/pytorch/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def parse_batchnorm_layer(pytorch_layer, layer_name, input_shapes, data_reader,
6565

6666
# batchnorm para
6767
layer['epsilon'] = pytorch_layer.eps
68-
layer['use_gamma'] = layer['use_beta'] = not pytorch_layer.affine
68+
layer['use_gamma'] = layer['use_beta'] = pytorch_layer.affine
6969

7070
in_size = 1
7171
for dim in input_shapes[0][1:]:

0 commit comments

Comments
 (0)