maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
640 commits 6 branches 0 tags 118 MiB
Commit graph

225 commits

Author SHA1 Message Date
gkarch 684678086a fixed speedup factor 2019-12-17 16:51:22 +01:00
gkarch 82518bad2d fixed speedup in trt readme 2019-12-17 16:49:44 +01:00
gkarch 5dbb6c5b91 updated perf scripts 2019-12-17 16:48:08 +01:00
gkarch 3634315318 fixed tabs 2019-12-16 18:31:29 +01:00
gkarch 2992264d3a updated Tacotron2: included trt, new dllogger 2019-12-16 18:29:08 +01:00
nv-kkudrynski e0b365ce6e
Merge pull request #347 from NVIDIA/nvpstr/63cf4f 
Nvpstr/63cf4f
 2019-12-16 13:24:07 +01:00
Hyungon_Ryu fb2709e002 [Transformer-XL/PyT] Update train.py (#348) 
bug fix for inv_sqrt scheduler
 2019-12-16 10:06:42 +01:00
Przemek Strzelczyk 5562ab767a Adding SE-ResNext and ResNext / PyT 2019-12-15 05:13:59 +01:00
Chris Forster 67b7543feb
Update README.md 
Adding link to our Medium.com article that provides details about implementing the LAMB optimizer.
 2019-12-13 12:51:41 -08:00
skierat 2fb7c8f472
Merge pull request #149 from HanbumKo/HanbumKo-patch-1 
Update inference.py
 2019-12-09 11:32:36 +01:00
kkudrynski c9c3eaacf1 minor fixes: submodules, jasper readme 2019-12-06 20:38:09 +01:00
Przemek Strzelczyk e89dfbf536 [Jasper/PyT] Fixing notebooks/README.md 2019-12-06 14:52:20 +01:00
Przemek Strzelczyk 09622fa363 [Jasper/PyT] Added: inference support for TRT6 and TRT-IS with various backends; new Jupyter notebooks 2019-12-05 20:40:27 +01:00
PrzemekS dc63c016cf
Merge pull request #312 from sharathts/patch-6 
Fix case with one training shard only
 2019-12-04 15:30:13 +01:00
Przemek Strzelczyk ca28f55476 [Transformer-XL/PyT] renaming folders 2019-11-28 09:48:59 +01:00
Przemek Strzelczyk 547bd323a0 [Jasper/PyT] Adding Colab TRT notebook 2019-11-27 17:07:31 +01:00
Przemek Strzelczyk 3d46067af9 Adding TransformerXL/PyT 2019-11-27 17:00:18 +01:00
Sharath T S 657874ae09
Fix case with one training shard only 2019-11-20 10:52:32 -08:00
nvpstr 7d772b8bc9
Merge pull request #299 from cschaefer26/tacotron2-fix-val-lo 
Fixed wrong val loss being logged
 2019-11-19 10:16:21 +01:00
nvpstr 3d4cc84640
Merge pull request #297 from GrzegorzKarchNV/tacotron2-readme-update 
updated tacotron2 trtis readme
 2019-11-19 10:08:52 +01:00
nvpstr fdf7124915
Merge pull request #71 from vinhngx/patch-1 
Include jupyter in this docker build
 2019-11-19 10:04:16 +01:00
nvpstr ad4267cb70
Merge pull request #69 from vinhngx/jupyter 
add this repo to docker build
 2019-11-19 10:03:46 +01:00
Przemek Strzelczyk a70896405d Updating BERT/PyT 
* Use LAMB from APEX
* Code cleanup
* Bug fix in BertAdam optimizer
 2019-11-18 23:07:24 +01:00
Christian Schäfer f0eede6bf8
Update train.py 
fixed logged val loss
 2019-11-15 09:46:46 +01:00
gkarch 09706537d2 updated tacotron2 trtis readme 2019-11-14 15:24:14 +01:00
GrzegorzKarchNV 4e00153ab5 added TRTIS demo to Tacotron2 (#281) 
* added TRTIS demo

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md
 2019-11-06 19:46:35 +01:00
Sharath T S 78e97e324e
Fix incorrect perf numbers 2019-10-25 15:52:44 -07:00
Yang Zhang 81df05da1b update Jasper TRT Readme numbers (#267) 
fix Readme table column name mix up
 2019-10-23 12:38:43 +02:00
nvpstr 970a54b296
Merge pull request #261 from sharathts/patch-3 
fix single gpu support
 2019-10-21 19:59:14 +02:00
Przemek Strzelczyk b09ce9831b Updating GNMT/PyT 2019-10-21 19:41:32 +02:00
Przemek Strzelczyk e470c2150a Updating RN50/MxNet 2019-10-21 19:20:40 +02:00
Sharath T S f24491a940
fix single gpu support 2019-10-18 17:28:02 -07:00
Sharath T S 7121e21d11
fix logging total steps 2019-10-16 15:12:00 -07:00
Sharath T S 8cc635f638
Fix training perf calculation 2019-10-15 17:39:39 -07:00
Przemek Strzelczyk 81519d47a2 NCF/PyT - adjusting to API changes in PyTorch and APEX 2019-10-04 15:05:13 +02:00
Przemek Strzelczyk d29920a8b0 MaskRCNN/PyT update 
* Jypyter notebooks addded
* Updates for PyTorch 1.2
 2019-10-04 14:34:26 +02:00
Przemek Strzelczyk 68c6d2321f Small updates to README.md (Jasper,notebooks) 2019-10-04 14:11:10 +02:00
maggiezha 075bca5742 update readme of notebook (#229) 
* updating notebook
* updating wav files
* updating examples
 2019-10-01 19:41:26 +02:00
Sharath T S 7fffeade4e
Update Dockerfile 2019-09-25 11:21:15 -07:00
Sharath T S 746cfe21dc
Update Dockerfile 2019-09-25 11:17:40 -07:00
Przemek Strzelczyk 10e805921a [Jasper/PyT] Updated notebooks 2019-09-23 16:08:53 +02:00
nvpstr 2fec868452
Merge pull request #221 from maggiezha/patch-2 
update readme
 2019-09-20 18:06:03 +02:00
nvpstr 85aa84467f
Merge pull request #220 from maggiezha/patch-1 
Update README.md
 2019-09-20 18:05:30 +02:00
yzhang123 5e126c7154
Update JasperTRT.ipynb 2019-09-19 11:02:01 -07:00
yzhang123 4db4d43829
Update JasperTRT.ipynb 
updated default mount directory paths
 2019-09-19 09:50:58 -07:00
maggiezha b13bcc9405
update readme 
running notebook directly from /notebooks/ doesn't work, the later container mapping went messy and gave me error like: No such file or directory, need to copy the notebook to root directory to work
 2019-09-19 15:39:55 +10:00
maggiezha c13e5f6704
Update README.md 
move requirements from notebook to readme
 2019-09-19 13:36:36 +10:00
maggiezha 6cbd0fa2a4
Update README.md 2019-09-19 13:31:17 +10:00
Przemek Strzelczyk f7b0a9c583 [Jasper/PyT] Small README and notebook fixes 2019-09-18 23:38:40 +02:00
Przemek Strzelczyk 2de99b5fa7 [Jasper/PyT] Adding TRT support + jupyter notebooks for inference 2019-09-18 22:05:24 +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 8b249efad6 Minor fixes to BERT/PyT 2019-09-13 15:23:39 +02:00
Przemek Strzelczyk 4cce4d88e6 Updating SSD/PyT 2019-09-12 14:33:49 +02:00
Przemek Strzelczyk 6fe463fe27 [BERT/PyT] Support for multi-node 2019-09-10 17:21:52 +02:00
Przemek Strzelczyk 02b49acead [Tacotron2] Added denoiser and inference stats, fixed typos 2019-09-10 16:22:53 +02:00
gkarch b8027d8914 added jupyter notebook 2019-09-03 18:40:29 +02:00
Chris Forster 71e2b22d4a Update bertPrep.py (#183) 2019-08-29 21:49:02 +02:00
Chris Forster e72ea6947b BERT-PyT subprocess for bzip in wikidownloader (#180) 
* Removing unnecessary subprocess.communicate calls

* Updating Bookscorpus downloader to require less memory

* Renaming variable
 2019-08-29 07:21:53 +02:00
Chris Forster 3d3ff3e168 Cleanup and Readme Update (#174) 
* update perf tables

* remove ide files

* fix tokenizer

* copyrights

* remove .communicate()

* refine training scripts

* fix more typos
 2019-08-27 21:44:21 +02:00
Sharath T S 3d59216cec [BERT] [PyTorch] Data prep fix (#171) 
* add dgx1-16g and dgx2 specific pretraining instructions

* fix typo in readme

* fix data prep and reflect changes in pretraining

* remove .ide files

* remove data files

* Point to right SQUAD location

* remove garbage [[]]

* default accumulation in fp32

* remove ide files

* fix phase2 DATADIR path

* remove readme in data folder
 2019-08-22 07:52:18 +02:00
Sharath T S b6fb9aa463 [BERT][PyTorch]: add dgx1-16g and dgx2 specific pretraining instructions (#164) 
* add dgx1-16g and dgx2 specific pretraining instructions

* fix typo in readme
 2019-08-21 09:49:32 +02:00
yzhang123 f84446675e
novograd default parameter fix 2019-08-16 14:07:05 -07:00
yzhang123 5419463c91
fix novograd 2019-08-16 14:05:18 -07:00
nv-kkudrynski 9f7616dc54
minor readme fix 2019-08-14 13:30:37 +02:00
Cliff Woolley b7bf42d76c
Update README.md 
Fix typo
 2019-08-13 16:12:52 -07:00
Cliff Woolley 608663f6ec Don't omit the data/ scripts from docker build 2019-08-13 15:41:48 -07:00
Cliff Woolley 8546c7a6df Cleanups 2019-08-13 15:33:32 -07:00
Cliff Woolley 7afcd73af1 Cleanup unneeded files 2019-08-13 15:32:00 -07:00
Krzysztof Kudrynski bae6e931bd updating BERT (single node LAMB support) 2019-08-13 23:27:54 +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
HanbumKo 8d365d4b1f
Update inference.py 
(mean, std) isn't used in load_image but normalize.
 2019-08-06 14:30:04 -07:00
sharatht 803963408a remove directory check in data download 2019-08-02 22:28:02 -07:00
nvpstr d9f925cb9c
Merge pull request #130 from NVIDIA/nvpstr/release 
Adding Jasper/PyT
 2019-07-30 16:18:27 +02:00
gkarch b56f72bcd9 fixed audio 2019-07-30 15:01:15 +02:00
Przemek Strzelczyk fa400a7367 Adding Jasper/PyT 2019-07-26 20:08:16 +02:00
nvpstr 6c0b5e36b4
Merge pull request #125 from NVIDIA/nvpstr/release 
Updating BERT with TRT-IS support and new results
 2019-07-25 20:47:32 +02:00
nvpstr d3813fde80
Merge pull request #122 from GrzegorzKarchNV/update19.07 
updated Tacotron2 version
 2019-07-25 18:46:33 +02:00
Przemek Strzelczyk 8218872051 Updating BERT with TRT-IS support and new results 2019-07-25 16:53:05 +02:00
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
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
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
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
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 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
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
Vinh Nguyen 7d3d1a028b
Include jupyter in this docker build 2019-06-05 18:38:36 +10:00
Vinh Nguyen 1cac44e014 add this repo to docker build 2019-06-05 16:32:58 +10: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
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
Vinh Nguyen 8662db35a7 fix readme: direct link for Apex. Fix download script location 2019-05-24 14:47:43 +10:00
Vinh Nguyen eee0490810 adding resume capability for wget in case downloads stop early 2019-05-24 14:41:23 +10:00
Krzysztof Kudrynski 14fe91fad5 added missing nested-modules-discovery from global entrypoint like torch.hub 2019-05-21 13:32:36 +02:00
Krzysztof Kudrynski 49acab486b torch.hub entrypoints for tacotron2, waveglow and ncf 2019-05-15 14:16:46 +02:00
skierat 2409559c12
Update Dockerfile 2019-04-01 14:34:29 +02:00
sharatht f1661d8ca4 Fix README and Dockerfile 2019-03-18 18:06:48 -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
Tomasz Grel 7c10b43d3e update NCF README 2019-02-21 14:54:47 +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
nvpstr a1aff31c42 Adding training improvements. 2019-01-25 16:02:48 +01:00
Przemek Strzelczyk edc3486e6f Small fix to README 2019-01-23 17:08:35 +01:00
Przemek Strzelczyk d755a84080 Adding Transformer to PyTorch/Translation 2019-01-23 17:03:45 +01:00
Przemek Strzelczyk 585973766f Adding NCF to PyTorch/Recommendation 2019-01-23 16:59:07 +01:00
nvpstr 3ac7dc4957 Adding ResNet50 v1.5. 2018-10-01 22:51:24 +02:00
Szymon Migacz 7d69e136a3 Added PyTorch GNMT implementation 2018-08-07 16:27:43 +02:00
Christian Sarofeen 76a3d10a19 Add pytorch examples 2018-05-02 18:33:56 -07:00