Commit graph

8 commits

Author SHA1 Message Date
Przemek Strzelczyk
15807b36bf Adding DLRM/PyT 2020-04-08 18:17:57 +02:00
Przemek Strzelczyk
1def26d80c [NCF/PyT] Adding new logging 2020-02-05 21:58:04 +01:00
Przemek Strzelczyk
81519d47a2 NCF/PyT - adjusting to API changes in PyTorch and APEX 2019-10-04 15:05:13 +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
Krzysztof Kudrynski
49acab486b torch.hub entrypoints for tacotron2, waveglow and ncf 2019-05-15 14:16:46 +02:00
Tomasz Grel
7c10b43d3e update NCF README 2019-02-21 14:54:47 +01:00
Przemek Strzelczyk
585973766f Adding NCF to PyTorch/Recommendation 2019-01-23 16:59:07 +01:00