NGC Catalog
CLASSIC
Welcome Guest
Models
GNMT 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
Latest Version
19.08.0_amp
Modified
April 4, 2023
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:

  • WMT16 English-German - Dataset for machine translation.

Performance

Performance numbers for this model are available in NGC.

References

  • Original paper
  • NVIDIA model implementation in NGC
  • NVIDIA model implementation on GitHub

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.