Skip to content

Commit 2afae66

Browse files
committed
fix quotation marks
1 parent 91ee88e commit 2afae66

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

hls4ml/converters/pytorch/pooling.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,7 @@ def parse_pooling_layer(operation, layer_name, input_names, input_shapes, node,
8686
padding = [class_object.padding, class_object.padding]
8787

8888
else:
89+
print(node.args)
8990
if type(node.kwargs['stride']) is tuple:
9091
layer['stride_height'] = node.kwargs['stride'][0]
9192
layer['stride_width'] = node.kwargs['stride'][1]

hls4ml/converters/pytorch/reshape.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ def parse_flatten_layer(operation, layer_name, input_names, input_shapes, node,
9393
layer['class_name'] = 'Reshape'
9494
layer['name'] = layer_name
9595
layer['inputs'] = input_names
96-
if node.op == "call_module":
96+
if node.op == 'call_module':
9797
start_dim = class_object.start_dim
9898
end_dim = class_object.end_dim
9999
if end_dim + 1 == 0 or end_dim + 1 > len(input_shapes[0]):

0 commit comments

Comments
 (0)