Commit graph

14 commits

Author SHA1 Message Date
Krzysztof Kudrynski 878b004e6b [NCF/TF, WideAndDeep/TF] Updating for Ampere, [BERT/TF] MRPC support 2020-07-14 12:34:25 +02:00
Przemek Strzelczyk 15ba45666d [WideAndDeep] Improved Spark preprocessing scripts performance 2020-05-18 01:11:27 +02:00
Przemek Strzelczyk d996f54542 [Wide&Deep/TF] README updates; results for 50 runs and small fixes 2020-04-08 15:12:25 +02:00
Przemek Strzelczyk 85d346c0b4 [WideAndDeep] Fixes 2020-03-30 15:44:33 +02:00
Przemek Strzelczyk 639de18491 Adding WideAndDeep/TF 2020-03-24 23:05:23 +01:00
Przemek Strzelczyk 43aa1de260 [NCF/TF] Adding new logging 2020-02-28 12:48:22 +01:00
Tomasz Grel a5976cb5a3 VAE-CF README fixes (#332) 2019-12-04 13:23:42 +01:00
Przemek Strzelczyk 678c71b989 [VAE-CF/TF] README fix 2019-12-02 17:59:43 +01:00
Przemek Strzelczyk b348f179e2 Adding VAE-CF/TF 2019-12-02 14:26:38 +01: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 d2bc3da0a1 Changes in TF models:
* added UNet for medical image segmentation
* added TF-AMP support for RN50
* small updates for other models (READMEs, benchmark & testing scripts)
2019-05-25 01:23:11 +02:00
Michał Krzysztof Filipiuk 041c35247b Fixed indentation 2019-03-25 12:52:08 -07: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