NGC | Catalog
Welcome Guest
CatalogResourcesNemo with Pytorch Lightning

Nemo with Pytorch Lightning

For downloads and more information, please view on a desktop device.
Logo for Nemo with Pytorch Lightning

Description

The primary objective of NeMo is to help researchers from industry and academia to reuse prior work and make it easier to create new conversational AI models.

Publisher

NVIDIA

Use Case

Other

Framework

Other

Latest Version

1

Modified

September 1, 2021

Compressed Size

1.78 MB

Nemo with PyTorch Lightning

PyTorch Lightning is a light weight PyTorch wrapper that reduces the engineering boilerplate and resources required to implement state-of-the-art AI. Organizing PyTorch code with Lightning, enables seamless training on multiple-GPUs, TPUs, CPUs as well as the use of difficult to implement best practices such as model sharding, 16-bit precision and more. PyTorch Lightning support an extremly robust ecosystem of Machine Learning and Deep Learning projects from 1st Party Repos such as Flash to 3rd party frameworks such as Nvidia NeMo. NVIDIA NeMo is a conversational AI toolkit built for researchers working on automatic speech recognition (ASR), natural language processing (NLP), and text-to-speech synthesis (TTS). The primary objective of NeMo is to help researchers from industry and academia to reuse prior work (code and pretrained models and make it easier to create new conversational AI models.

Run the Jupyter notebook:

Download the jupyter notebook from top right section of this page and run the cells of the notebook to install required libraries, download the models and train the model.