Skip to content

Commit 229dc7b

Browse files
committed
another fix to pytests for upsampling
1 parent 88e1f9b commit 229dc7b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/pytest/test_upsampling_pytorch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def test_pytorch_upsampling1d(data_1d, io_type, backend):
5656
config = hls4ml.utils.config_from_pytorch_model(
5757
model,
5858
default_precision='ap_fixed<16,6>',
59-
channels_last_conversion="off", # We don't test channels_last conversion here
59+
channels_last_conversion="internal",
6060
transpose_outputs=False,
6161
)
6262
odir = str(test_root_path / f'hls4mlprj_pytorch_upsampling_1d_{backend}_{io_type}')
@@ -85,7 +85,7 @@ def test_pytorch_upsampling2d(data_2d, io_type, backend):
8585
config = hls4ml.utils.config_from_pytorch_model(
8686
model,
8787
default_precision='ap_fixed<16,6>',
88-
inputs_channel_last=False, # With conversion to channels_last
88+
channels_last_conversion="full", # With conversion to channels_last
8989
transpose_outputs=True,
9090
)
9191
odir = str(test_root_path / f'hls4mlprj_pytorch_upsampling_2d_{backend}_{io_type}')

0 commit comments

Comments
 (0)