Skip to content

Commit

Permalink
docs: Update Tensorflow and Pytorch documentation
Browse files Browse the repository at this point in the history
This PR updates the Tensorflow and Pytorch documentation that we have
for kata containers metrics.

Fixes kata-containers#5679

Signed-off-by: Gabriela Cervantes <gabriela.cervantes.tellez@intel.com>
  • Loading branch information
GabyCT committed Jun 2, 2023
1 parent 1188a01 commit 6a3f825
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions metrics/machine_learning/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand All @@ -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

Expand Down

0 comments on commit 6a3f825

Please sign in to comment.