This directory contains Jupyter notebooks that demonstrate model compilation and inference using TensorFlow Neuron for a variety of popular deep learning models. These samples can be run on AWS Inferentia (inf1 instances) using Amazon SageMaker or Amazon EC2.
For each sample you will also find additional information such as the model type, configuration used to compile the model, framework version, and a link to the original model implementation.
The following samples are available:
| Model Name | Model Type | Input Shape | NeuronSDK Version | Framework / Version | Original Implementation |
|---|---|---|---|---|---|
| U-Net | CV - Semantic Segmentation | 1,3,224,224 | 2.5.2.2.1.14.0 | Tensorflow 2.5.2 | link |
In order to run the samples, you first need to set up a TensorFlow Neuron development environment.