diff --git a/tests/metrics/gha-run.sh b/tests/metrics/gha-run.sh index 3f168dafec..576e8273e1 100755 --- a/tests/metrics/gha-run.sh +++ b/tests/metrics/gha-run.sh @@ -80,7 +80,7 @@ function run_test_blogbench() { function run_test_tensorflow() { info "Running TensorFlow test using ${KATA_HYPERVISOR} hypervisor" - bash tests/metrics/machine_learning/tensorflow.sh 1 20 + bash tests/metrics/machine_learning/tensorflow_nhwc.sh 1 20 check_metrics }