NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
QuartzNet for PyTorch
Resource
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
QuartzNet for PyTorch

End-to-end neural acoustic model for automatic speech recognition providing high accuracy at a low memory footprint.

The following section lists the requirements that you need to meet in order to start training the QuartzNet model.

Requirements

This repository contains Dockerfile which extends the PyTorch 21.07-py3 NGC container and encapsulates some dependencies. Aside from these dependencies, ensure you have the following components:

For more information about how to get started with NGC containers, see the following sections from the NVIDIA GPU Cloud Documentation and the Deep Learning Documentation:

Further required Python packages are listed in requirements.txt, which are automatically installed with the built Docker container. To manually install them, run:

pip install -r requirements.txt

For those unable to use the PyTorch 21.07-py3 NGC container, to set up the required environment or create your own container, see the versioned NVIDIA Container Support Matrix.