Commit graph

183 commits

Author SHA1 Message Date
yzhang123 0af34d778c
fix launch.sh 2019-07-24 12:24:43 -07:00
yzhang123 2eb764b43c
fix build.sh 2019-07-24 12:23:34 -07:00
Grzegorz Karch d0c6294695 removed cudnn-benchmark from tacotron2 perf command lines in the readme 2019-07-24 01:08:20 -07:00
Grzegorz Karch 7e5013fff9 fixed script names in the readme 2019-07-24 00:30:07 -07:00
Grzegorz Karch bb7a4ac630 updated container version in the readme, changed order of benchmarking scripts 2019-07-23 15:50:02 -07:00
Grzegorz Karch 6c42c20948 updated readme, number of epochs in training scripts 2019-07-23 15:43:24 -07:00
Grzegorz Karch 87accc3073 updated performance benchmark command lines in the readme 2019-07-23 15:26:54 -07:00
Grzegorz Karch 25b3f2678a fixed inference results 2019-07-23 14:29:58 -07:00
Grzegorz Karch 979e291848 updated to 19.07 version 2019-07-23 12:45:37 -07:00
nvpstr 3137fbeae3
Merge pull request #120 from NVIDIA/nvpstr-patch-2
Reverting changes to Dockerfile
2019-07-22 16:14:09 +02:00
nvpstr a230e06917
Update Dockerfile 2019-07-22 16:07:25 +02:00
xjia 5777ab97b7 delete duplicate files 2019-07-20 00:17:18 +08:00
xjia 9f00666591 support arbitrary sequence 2019-07-19 21:23:57 +08:00
Marek Wawrzos c5829f0b44 use dataset preprocessing script from tensorflow/object_detection 2019-07-17 21:56:41 +02:00
nvpstr 81b1b024e1
Update Dockerfile 2019-07-16 21:57:29 +02:00
nvpstr f6c367e967
Merge pull request #107 from NVIDIA/nvpstr/master3
Removing flawed Transformer
2019-07-16 21:34:02 +02:00
Przemek Strzelczyk fde2be5bb6 Removing flawed Transformer/TF 2019-07-16 21:30:03 +02:00
nvpstr 8c35140712
Merge pull request #106 from NVIDIA/nvpstr/master3
Updating models and adding BERT/PyT
2019-07-16 21:27:09 +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 3b3d0f6a55 Merge branch 'gh/master' into nvpstr/master3 2019-07-16 19:56:26 +02:00
Michael O'Connor 2cfd880a6b
Merge pull request #103 from lxp121/master
Adding FasterTransformer
2019-07-13 09:07:49 -10:00
Xipeng Li 75502be814 Adding FasterTransformer: A faster transformer layer inference implementation for BERT and other transformer based models. 2019-07-14 00:29:45 +08:00
Przemek Strzelczyk 0663b67c1a Updating models 2019-07-08 22:51:28 +02:00
nvpstr f89dcca19d
Merge pull request #89 from GrzegorzKarchNV/readme-epochs
changed number of epochs in train scripts; removed number of epochs f…
2019-07-01 11:44:01 +02:00
nvpstr eabbedd1d4
Merge pull request #91 from GrzegorzKarchNV/readme-fix
fixed batches for mixed precision and fp32, added info when no input …
2019-07-01 11:43:29 +02:00
Grzegorz Karch d7c20e5bfc fixed batches for mixed precision and fp32, added info when no input is provided for inference, added units to expected training time 2019-06-27 01:48:46 -07:00
Grzegorz Karch 7eb5eece14 changed number of epochs in train scripts; removed number of epochs from readme for the audio samples 2019-06-26 08:57:23 -07:00
nvpstr 40e074257f
Merge pull request #65 from JanuszL/fix_RN50_pytorch_DALI
Fix typo in RN50 PyTorch readme, remove unneeded lines for DALI
2019-06-13 17:59:16 +02:00
nvpstr 34a01f7b72
Merge pull request #77 from GrzegorzKarchNV/warmup
updated readme: text.txt->phrase.txt, updated inference results
2019-06-13 17:35:33 +02:00
nvpstr 9df04c29c9
Merge pull request #79 from nv-kkudrynski/ssd_torchhub_exposure
ssd exposure via hubconf.py (together with ssd utils)
2019-06-13 17:35:16 +02:00
Krzysztof Kudrynski a8328ce169 ssd exposure via hubconf.py (together with ssd utils) 2019-06-12 17:41:57 +02:00
Grzegorz Karch cd284a62ec changed text to phrase as input to inference in the readme 2019-06-07 17:37:30 +02:00
Grzegorz Karch dc7cd1ae5b updated readme 2019-06-07 17:37:30 +02:00
gkarch 79b9a48bf1 added warmup parameter to tacotron2 inference; updated README 2019-06-07 17:37:30 +02:00
nvpstr 4aa6f17167
Merge pull request #76 from GrzegorzKarchNV/torchhub
added text_to_sequence in the hubconf model
2019-06-07 15:11:28 +02:00
gkarch 362dfe6b3b added text_to_sequnce in the hubconf model 2019-06-07 14:52:23 +02:00
nvpstr da165c3a52
Merge pull request #73 from GrzegorzKarchNV/master
updated readme - warmup for inference, updated results for 19.05-py3 container
2019-06-06 16:03:19 +02:00
nvpstr bf1b29d73e
Merge pull request #63 from khcs/bert/pretrain-fix
Fix FLAG.save_checkpoint_steps naming inconsistency.
2019-06-06 16:02:28 +02:00
nvpstr 698e775680
Merge pull request #72 from jbaczek/master
Add missing submodule for PyT Transformer
2019-06-06 16:01:51 +02:00
nvpstr c5bbd74f87
Merge pull request #62 from NVIDIA/nvpstr-patch-1
Update README.md
2019-06-06 15:58:28 +02:00
gkarch 2f34442146 update perf numbers 2019-06-06 15:27:21 +02:00
gkarch 4dd9ecfcfa updated readme - warmup for inference 2019-06-06 14:16:51 +02:00
jbaczek 2774e849be remove empty dir 2019-06-05 16:20:30 +02:00
jbaczek 743162906c fix submodule path 2019-06-05 16:17:24 +02:00
jbaczek 698516f8f7 add missing submodule 2019-06-05 16:14:41 +02:00
nvpstr 2619f172c7
Merge pull request #67 from GrzegorzKarchNV/master
Update models.py - fix fp16 inference
2019-06-05 14:45:27 +02:00
GrzegorzKarchNV 867ab68fad
Update models.py 2019-06-04 15:44:05 +02:00
Janusz Lisiecki 20f09772f8 Fix typo in RN50 PyTorch readme, remove unneeded lines for DALI
Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
2019-06-03 12:03:18 +02:00
Hoo Chang Shin f3c1cc7204 Fix FLAG.save_checkpoint_steps naming inconsistency. 2019-05-30 11:37:46 -04:00
nvpstr e5a0a056d2
Update README.md 2019-05-30 13:42:31 +02:00