Skip to content

Commit d54843d

Browse files
authored
Merge pull request #663 from fastmachinelearning/jmitrevs/pytest_cnn_fix
the strategy was not propagated in the pytest
2 parents 90d760a + 6219dad commit d54843d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/pytest/test_cnn_mnist_qkeras.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def mnist_model():
4545
def hls_model(mnist_model, backend, io_type, strategy):
4646
keras_model = mnist_model
4747
hls_config = hls4ml.utils.config_from_keras_model(keras_model, granularity='name')
48-
hls_config['Model']['Strategy'] = 'Resource'
48+
hls_config['Model']['Strategy'] = strategy
4949
hls_config['LayerName']['softmax']['Strategy'] = 'Stable'
5050
output_dir = str(test_root_path / 'hls4mlprj_cnn_mnist_qkeras_{}_{}_{}'.format(backend, io_type, strategy))
5151

0 commit comments

Comments
 (0)