NGC | Catalog
CatalogModelsGNMT TensorFlow checkpoint (AMP)

GNMT TensorFlow checkpoint (AMP)

For downloads and more information, please view on a desktop device.
Logo for GNMT TensorFlow checkpoint (AMP)

Description

GNMT TensorFlow checkpoint trained with AMP

Publisher

NVIDIA Deep Learning Examples

Use Case

Translation

Framework

TensorFlow

Latest Version

19.08.0

Modified

October 29, 2021

Size

1.84 GB

Model Overview

The GNMT v2 model is an improved version of the first Google's Neural Machine Translation System with a modified attention mechanism.

Model Architecture

The following image shows the GNMT model architecture:

TrainingLoss

Training

This model was trained using script available on NGC and in GitHub repo

Dataset

The following datasets were used to train this model:

Performance

Performance numbers for this model are available in NGC

References

License

This model was trained using open-source software available in Deep Learning Examples repository. For terms of use, please refer to the license of the script and the datasets the model was derived from.