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