mirror of
https://github.com/kata-containers/kata-containers.git
synced 2025-07-19 01:39:48 +00:00
c8b43f8b3e
This PR fixes the pytorch reference in the README file. Fixes #7689 Signed-off-by: Gabriela Cervantes <gabriela.cervantes.tellez@intel.com>
63 lines
2.0 KiB
Markdown
63 lines
2.0 KiB
Markdown
# Kata Containers TensorFlow Metrics
|
|
|
|
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 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
|
|
|
|
Individual tests can be run by hand, for example:
|
|
|
|
```
|
|
$ cd metrics/machine_learning
|
|
$ ./tensorflow_nhwc.sh 25 60
|
|
```
|
|
# Kata Containers Pytorch Metrics
|
|
|
|
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
|
|
|
|
Individual tests can be run by hand, for example:
|
|
|
|
```
|
|
$ cd metrics/machine_learning
|
|
$ ./pytorch.sh 40 100
|
|
```
|
|
# Kata Containers TensorFlow `MobileNet` Metrics
|
|
|
|
`MobileNets` are small, low-latency, low-power models parameterized to meet the resource
|
|
constraints of a variety of use cases. They can be built upon for classification, detection,
|
|
embeddings and segmentation similar to how other popular large scale models, such as Inception, are used.
|
|
`MobileNets` can be run efficiently on mobile devices with `Tensorflow` Lite.
|
|
|
|
Kata Containers provides a test for running `MobileNet V1` inference using Intel-Optimized `TensorFlow`.
|
|
|
|
## Running the `TensorFlow` `MobileNet` test
|
|
Individual test can be run by hand, for example:
|
|
|
|
```
|
|
$ cd metrics/machine_learning
|
|
$ ./tensorflow_mobilenet_benchmark.sh 25 60
|
|
```
|
|
|
|
# Kata Containers TensorFlow `ResNet50` Metrics
|
|
|
|
`ResNet50` is an image classification model pre-trained on the `ImageNet` dataset.
|
|
Kata Containers provides a test for running `ResNet50` inference using Intel-Optimized
|
|
`TensorFlow`.
|
|
|
|
## Running the `TensorFlow` `ResNet50` test
|
|
Individual test can be run by hand, for example:
|
|
|
|
```
|
|
$ cd metrics/machine_learning
|
|
$ ./tensorflow_resnet50_int8.sh 25 60
|
|
```
|