NGC Catalog
CLASSIC
Welcome Guest
Models
GNMT PyTorch checkpoint (AMP)

GNMT PyTorch checkpoint (AMP)

For downloads and more information, please view on a desktop device.
Logo for GNMT PyTorch checkpoint (AMP)
Description
GNMT PyTorch checkpoint trained with AMP
Publisher
NVIDIA Deep Learning Examples
Latest Version
19.08.0
Modified
April 4, 2023
Size
1.5 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

ModelArchitecture

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.