Commit graph

10 commits

Author SHA1 Message Date
kkudrynski 5d2914e3ff [GNMT/PyT] Updating for Ampere] 2020-08-01 15:47:34 +02:00
Przemek Strzelczyk b09ce9831b Updating GNMT/PyT 2019-10-21 19:41:32 +02:00
Szymon Migacz 3014f38a3f
[GNMT PyT] Fix for fp16 training w/o label smoothing (#210) 2019-09-16 10:09:28 +02:00
Przemek Strzelczyk a644350589 Updating models and adding BERT/PyT
Tacotron2+Waveglow/PyT
* AMP support
* Data preprocessing for Tacotron 2 training
* Fixed dropouts on LSTMCells

SSD/PyT
* script and notebook for inference
* AMP support
* README update
* updates to examples/*

BERT/PyT
* initial release

GNMT/PyT
* Default container updated to NGC PyTorch 19.05-py3
* Mixed precision training implemented using APEX AMP
* Added inference throughput and latency results on NVIDIA Tesla V100 16G
* Added option to run inference on user-provided raw input text from command line

NCF/PyT
* Updated performance tables.
* Default container changed to PyTorch 19.06-py3.
* Caching validation negatives between runs

Transformer/PyT
* new README
* jit support added

UNet Medical/TF
* inference example scripts added
* inference benchmark measuring latency added
* TRT/TF-TRT support added
* README updated

GNMT/TF
* Performance improvements

Small updates (mostly README) for other models.
2019-07-16 21:13:08 +02:00
Przemek Strzelczyk 0663b67c1a Updating models 2019-07-08 22:51:28 +02:00
Przemek Strzelczyk 7a8544e0a2 Adding 9 new models: ResNet50/TF, SSD/TF, BERT/TF, NCF/TF, UNet/TF, GNMT/TF, SSD/PyT, MaskRCNN/PyT, Tacotron2+Waveglow/PyT 2019-03-18 19:45:05 +01:00
Szymon Migacz b050963444 GNMT update 2019-02-14 12:40:30 +01:00
Szymon Migacz b99d45b3f5 Revert "Updating GNMT"
This reverts commit 072523d5b2.
2019-02-14 12:39:20 +01:00
nvpstr 072523d5b2 Updating GNMT 2019-01-25 16:20:05 +01:00
Szymon Migacz 7d69e136a3 Added PyTorch GNMT implementation 2018-08-07 16:27:43 +02:00