Skip to content

Commit 88e1f9b

Browse files
committed
change upsampling test to changed interface for channels_last conversion
1 parent 5299392 commit 88e1f9b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/pytest/test_upsampling_pytorch.py

Lines changed: 1 addition & 1 deletion
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-
inputs_channel_last=True, # We don't test channels_last conversion here
59+
channels_last_conversion="off", # We don't test channels_last conversion here
6060
transpose_outputs=False,
6161
)
6262
odir = str(test_root_path / f'hls4mlprj_pytorch_upsampling_1d_{backend}_{io_type}')

0 commit comments

Comments
 (0)