From 8ccc1e5c939de2decff64e019b0ed87dd9381763 Mon Sep 17 00:00:00 2001 From: Gabriela Cervantes Date: Mon, 17 Jul 2023 16:17:39 +0000 Subject: [PATCH] metrics: Update machine learning documentation This PR updates the machine learning documentation related with Tensorflow and Pytorch benchmarks. Fixes #7359 Signed-off-by: Gabriela Cervantes --- tests/metrics/machine_learning/README.md | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/tests/metrics/machine_learning/README.md b/tests/metrics/machine_learning/README.md index 8c62b1566f..66d15b0595 100644 --- a/tests/metrics/machine_learning/README.md +++ b/tests/metrics/machine_learning/README.md @@ -3,7 +3,10 @@ Kata Containers provides a series of performance tests using the TensorFlow reference benchmarks (tf_cnn_benchmarks). The tf_cnn_benchmarks containers TensorFlow implementations of several -popular convolutional models. +popular convolutional models https://github.com/tensorflow/benchmarks/tree/master/scripts/tf_cnn_benchmarks. + +Currently the TensorFlow benchmark on Kata Containers includes test for +the `AxelNet` and `ResNet50` models. ## Running the test @@ -15,9 +18,9 @@ $ ./tensorflow.sh 25 60 ``` # Kata Containers Pytorch Metrics -Kata Containers provides a series of performance tests using Pytorch -benchmarks based on a suite of Python high performance computing -benchmarks. +Based on a suite of Python high performance computing benchmarks that +uses various popular Python HPC libraries using Python + https://github.com/dionhaefner/pyhpc-benchmarks. ## Running the Pytorch test