Skip to content

Commit c2ca3c6

Browse files
[pre-commit.ci] auto fixes from pre-commit hooks
1 parent d247057 commit c2ca3c6

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

hls4ml/converters/pytorch_to_hls.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,13 +17,11 @@ def __init__(self, config):
1717
def get_weights_data(self, layer_name, var_name):
1818
data = None
1919

20-
2120
# Workaround for naming schme in nn.Sequential,
2221
# have to remove the prefix we previously had to add to make sure the tensors are found
2322
if 'layer_' in layer_name:
2423
layer_name = layer_name.split('layer_')[-1]
2524

26-
2725
# if a layer is reused in the model, torch.FX will append a "_n" for the n-th use
2826
# have to snap that off to find the tensors
2927
if layer_name.split('_')[-1].isdigit() and len(layer_name.split('_')) > 1:

0 commit comments

Comments
 (0)