docs: Update Tensorflow and Pytorch documentation

This PR updates the Tensorflow and Pytorch documentation that we have for kata containers metrics. Fixes kata-containers#5679 Signed-off-by: Gabriela Cervantes <[email protected]>
GabyCT · Jun 2, 2023 · 6a3f825 · 6a3f825
1 parent 1188a01
commit 6a3f825
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/metrics/machine_learning/README.md b/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 Kata Containers Tensorflow benchmarks includes tests for
+AxelNet and Resnet50 models.
 
 ## Running the test
 
@@ -15,9 +18,8 @@ $ ./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
+varios popular Python HPC libraries using Python https://github.com/dionhaefner/pyhpc-benchmarks.
 
 ## Running the Pytorch test