mirror of
https://github.com/kata-containers/kata-containers.git
synced 2025-04-29 12:14:48 +00:00
metrics: Rename tensorflow scripts
This PR renames the tensorflow scripts to include the data format that is being used as we will have multiple tests with different data and model formats for tensorflow so this will help us to distinguish them. Fixes #7645 Signed-off-by: Gabriela Cervantes <gabriela.cervantes.tellez@intel.com>
This commit is contained in:
parent
a740c80251
commit
18a7fd8e4e
tests/metrics/machine_learning
@ -10,15 +10,15 @@ set -o pipefail
|
||||
SCRIPT_PATH=$(dirname "$(readlink -f "$0")")
|
||||
source "${SCRIPT_PATH}/../lib/common.bash"
|
||||
|
||||
IMAGE="docker.io/library/tensorflow:latest"
|
||||
DOCKERFILE="${SCRIPT_PATH}/tensorflow_dockerfile/Dockerfile"
|
||||
IMAGE="docker.io/library/tensorflow_nhwc:latest"
|
||||
DOCKERFILE="${SCRIPT_PATH}/tensorflow_nhwc_dockerfile/Dockerfile"
|
||||
BATCH_SIZE="100"
|
||||
NUM_BATCHES="100"
|
||||
resnet_tensorflow_file=$(mktemp resnettensorflowresults.XXXXXXXXXX)
|
||||
alexnet_tensorflow_file=$(mktemp alexnettensorflowresults.XXXXXXXXXX)
|
||||
NUM_CONTAINERS="$1"
|
||||
TIMEOUT="$2"
|
||||
TEST_NAME="tensorflow"
|
||||
TEST_NAME="tensorflow_nhwc"
|
||||
PAYLOAD_ARGS="tail -f /dev/null"
|
||||
# Options to control the start of the workload using a trigger-file
|
||||
dst_dir="/host"
|
||||
@ -99,7 +99,7 @@ function launch_workload() {
|
||||
done
|
||||
}
|
||||
|
||||
function tensorflow_test() {
|
||||
function tensorflow_nhwc_test() {
|
||||
# Resnet section
|
||||
info "Running TF-Resnet test"
|
||||
launch_workload "${CMD_RESNET}"
|
||||
@ -191,7 +191,7 @@ function main() {
|
||||
# Get the initial number of pids in a single container before the workload starts
|
||||
INITIAL_NUM_PIDS=$(sudo -E "${CTR_EXE}" t metrics "${containers[-1]}" | grep pids.current | grep pids.current | xargs | cut -d ' ' -f 2)
|
||||
((INITIAL_NUM_PIDS++))
|
||||
tensorflow_test
|
||||
tensorflow_nhwc_test
|
||||
metrics_json_save
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user