Commit graph

8 commits

Author SHA1 Message Date
Peter-1213 9cf6c3824b Fix a logic flaw that causes checkpoints to be saved at wrong epochs. 2021-10-27 01:22:33 +08:00
Jan Baczek 233287038c [Transformer/PyT] Removing obsolete profiling code 2021-08-13 16:36:58 +00:00
Jan Baczek 7e8102d0ec [Transformer/PyT] Update to 21.05 2021-06-30 11:50:08 +00:00
Przemek Strzelczyk 2860d6fe04 [Transformer/PyT] Updating for 20.06 and Ampere 2020-07-04 02:28:25 +02:00
Krzysztof Kudrynski ab85e6cc3d Updating Tacotron2_pyt (BatchNorm init fix), Resnet_tf (cosine LR),
Transformer_pyt (bugfix)
2019-08-13 15:01:10 +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 5eaebefcd0 Updating PyTorch models
* New training scripts & results for RN50 PyT
* Documentation updates for Trasformer, SSD and MaskRCNN
2019-05-27 15:24:14 +02:00
Przemek Strzelczyk d755a84080 Adding Transformer to PyTorch/Translation 2019-01-23 17:03:45 +01:00