Commit graph

142 commits

Author SHA1 Message Date
Swetha Mandava
59b89880e3 allow_growth_fix=True removed 2019-11-13 11:06:15 -08:00
Hoo Chang Shin
92e52384f2 fix minor typo in README. 2019-11-13 11:38:41 -05:00
Hoo Chang Shin
caece99944 fix minor typo on README. 2019-11-13 11:36:49 -05:00
Hoo Chang Shin
536a5cfffc fix minor typo on README. 2019-11-13 11:29:57 -05:00
Hoo Chang Shin
70c51ad7f8 Merge remote-tracking branch 'upstream/master' 2019-11-13 11:28:26 -05:00
Swetha Mandava
b2466a0004
mismatch columns in readme 2019-11-12 13:55:16 -08:00
Haibin Lin
1ae1592ef8
Update optimization.py 2019-11-11 10:11:37 -08:00
Swetha Mandava
e94c6936e4
Merge pull request #3 from lapolonio/patch-3
Update Dockerfile
2019-11-08 11:24:03 -08:00
Leonardo Apolonio
826fcd4d33
Update Dockerfile
add libcurl3 install required by python when importing tensorrtserver.api module
2019-11-08 10:17:58 -05:00
Leonardo Apolonio
46a3d6e1de
Update run_trtis.sh
default trtis_model_name in run_trtis.sh should be consistent with export_model.sh
2019-11-08 10:15:53 -05:00
Swetha Mandava
e873db23cd moving trtis to 19.08 2019-11-07 13:07:47 -08:00
nvpstr
4760c03a04
Merge pull request #277 from NVIDIA/nvpstr/release
Updating BERT/TF
2019-11-04 23:33:38 +01:00
Przemek Strzelczyk
eb32aee293 BERT/TF: Dockerfile comment removed 2019-11-04 23:30:06 +01:00
Przemek Strzelczyk
9cd3946603 Updating BERT/TF
- Pre-training and Finetuning on BioMedical tasks and corpus
- Disabling Grappler Optimizations for improved performance
2019-11-04 23:18:08 +01:00
dependabot[bot]
63861b0621
Bump pillow from 5.4.1 to 6.2.0 in /TensorFlow/Segmentation/UNet_Medical
Bumps [pillow](https://github.com/python-pillow/Pillow) from 5.4.1 to 6.2.0.
- [Release notes](https://github.com/python-pillow/Pillow/releases)
- [Changelog](https://github.com/python-pillow/Pillow/blob/master/CHANGES.rst)
- [Commits](https://github.com/python-pillow/Pillow/compare/5.4.1...6.2.0)

Signed-off-by: dependabot[bot] <support@github.com>
2019-11-02 05:00:20 +00:00
Przemek Strzelczyk
11f6038cd8 Updating RN50/TF 2019-10-21 19:49:00 +02:00
Przemek Strzelczyk
52c4acfb46 Updating GNMT/TF 2019-10-21 19:44:27 +02:00
Przemek Strzelczyk
882bc26d40 [UNet_Industrial/TF] Adding Jupyter notebooks and small fixes 2019-10-09 14:45:45 +02:00
nvpstr
4c385522d1
Merge pull request #246 from swethmandava/master
BERT TF syntactical fixes
2019-10-09 14:10:56 +02:00
Swetha Mandava
8749cf3952 don't drop remainder in eval, num_shards for data 2019-10-07 19:00:55 -07:00
Swetha Mandava
aac3e7b6e9 adding missing num accum steps to readme 2019-10-07 17:30:04 -07:00
Swetha Mandava
b85439958d fix logo display on github 2019-10-07 11:35:22 -07:00
Swetha Mandava
b13aeea8d3 readme column mismatch and extra config removed 2019-10-07 11:32:29 -07:00
nv-kkudrynski
66a447276e
Update readme with info about external ckpts (#244) 2019-10-04 15:56:15 +02:00
Swetha Mandava
888e01448e change folder 2019-09-25 17:39:36 -07:00
Swetha Mandava
f5d1a8ba23 adding colab notebook, Use new TF-TRT AP 2019-09-25 17:29:12 -07:00
Swetha Mandava
df118e7155 changelog 2019-09-23 13:17:16 -07:00
Swetha Mandava
fa42f0ac27 add fp16/fp32 speedup for inference 2019-09-20 13:26:32 -07:00
Swetha Mandava
71fea240de switch ordering in readme 2019-09-19 22:30:37 -07:00
Swetha Mandava
890fc1c143 fix data dir path in eval 2019-09-19 16:54:11 -07:00
Swetha Mandava
d6c5fef145 adding notebooks and t4 results, cleanup 2019-09-19 10:02:06 -07:00
Przemek Strzelczyk
a98df279fe [BERT/TF] Added multi-node support 2019-09-13 19:12:50 +02:00
nv-kkudrynski
e25c23e14a
Merge pull request #153 from tlkh/master
Update formatting in AMP demo notebook
2019-08-14 13:43:04 +02:00
Krzysztof Kudrynski
52138acee6 resnet_tf readme fix 2019-08-13 15:51:10 +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
Timothy Liu
ee64eb2d9b Update formatting in AMP demo notebook 2019-08-10 12:17:14 +08:00
jconwayNV
55a03b8b4e
Added newline formatting 2019-08-09 17:13:20 -07:00
jconwayNV
0ed7d8a1d9
Added TF API to summary section
https://www.tensorflow.org/api_docs/python/tf/train/experimental/enable_mixed_precision_graph_rewrite
2019-08-09 17:12:20 -07:00
Timothy Liu
564ddc8946 Added demo notebook for AMP (image classification) 2019-08-09 22:00:40 +08:00
Swetha Mandava
281beab09d
consistant parameter names
consistant parameter names in run_pretraining.py and optimization.py
2019-07-30 17:24:19 -07:00
Swetha Mandava
438c422813 moving trtis section in readme and github issue 131 2019-07-30 13:35:10 -07:00
Swetha Mandava
c15d09eed6 moving trtis section in readme and github issue 131 2019-07-30 13:27:07 -07:00
Swetha Mandava
2b3b299448 moving trtis section in readme and github issue 131 2019-07-30 13:18:12 -07:00
Przemek Strzelczyk
fa400a7367 Adding Jasper/PyT 2019-07-26 20:08:16 +02:00
nvpstr
71fdde7200
Merge pull request #126 from NVIDIA/nvpstr/release
Nvpstr/release
2019-07-25 21:06:52 +02:00
Przemek Strzelczyk
a3dba86d52 Updating .gitmodules 2019-07-25 20:58:38 +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
Przemek Strzelczyk
c3ec91c6b9 Reverting changes on run_pretrainng.py 2019-07-25 20:43:32 +02:00
Przemek Strzelczyk
7a0d4fb8be Merge branch 'nvpstr/release' of https://gitlab-master.nvidia.com/dl/JoC/examples into nvpstr/release 2019-07-25 20:16:36 +02:00
Przemek Strzelczyk
15075aab0c Removing obsolete scripts 2019-07-25 20:15:52 +02:00
nvpstr
7fec94c592
Merge pull request #112 from mwawrzos/master
[TensorFlow/Detection/SSD][issue #102] use dataset preprocessing script from tensorflow/object_detection
2019-07-25 18:47:05 +02:00
Przemek Strzelczyk
566f9cba25 Removing some unnecessary marks 2019-07-25 07:58:23 -07:00
Przemek Strzelczyk
785801d699 Fixing Feature Support Matrix in README.md 2019-07-25 07:56:32 -07:00
Przemek Strzelczyk
8218872051 Updating BERT with TRT-IS support and new results 2019-07-25 16:53:05 +02:00
nvpstr
a230e06917
Update Dockerfile 2019-07-22 16:07:25 +02: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
Przemek Strzelczyk
fde2be5bb6 Removing flawed Transformer/TF 2019-07-16 21:30:03 +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
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
Hoo Chang Shin
37c4083c4b change to BERT-base, and checkpoints-dir. 2019-05-31 10:31:26 -04:00
Hoo Chang Shin
1d8ac4b295 Merge branch 'bert/pretrain-fix' 2019-05-30 12:58:03 -04:00
Hoo Chang Shin
65bf7dbe48 initial commit on training PubMed-gvocab-base. 2019-05-30 12:55:57 -04: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
nvpstr
b7eff3da41
Merge pull request #60 from swethmandava/smandava/bug_fix
num_gpus pretraining fix
2019-05-30 13:25:58 +02:00
Swetha Mandava
b25181c9b8 num_gpus pretraining fix 2019-05-28 09:52:52 -07:00
nvpstr
12c7bc2302
Merge pull request #45 from eric-haibin-lin/patch-1
Fix a few typos, and add gluonnlp implementation link
2019-05-25 01:37:31 +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
nvpstr
3cf8d6c3c5
Merge pull request #47 from swethmandava/master
fix mismatch in cl arguments for BERT
2019-05-20 13:59:49 +02:00
Swetha Mandava
c5951be727
fix mismatch in cl arguments 2019-05-19 10:10:58 -07:00
Haibin Lin
e46eed4c57
Update README.md 2019-05-18 22:41:42 -07:00
Khoa Ho
b0d259264e
Typo/stylistic fix for README title 2019-05-14 10:30:27 -07:00
nvpstr
a9d0554e5d
Merge pull request #30 from swethmandava/smandava/readme_edit
changing perf numbers to steady state throughput
2019-05-13 17:32:30 +02:00
Swetha Mandava
5551752822
Update readme error 2019-04-18 11:51:46 -07:00
Swetha Mandava
83fceae509 changing perf numbers to steady state throughput 2019-04-17 14:16:40 -07:00
nvpstr
51d3dceb3f
Merge pull request #24 from swethmandava/smandava/readme_edit
fix readme
2019-04-08 18:18:39 +02:00
Swetha Mandava
ec9eaf732a removing fast_math, etc from README 2019-04-04 13:14:01 -07:00
Swetha Mandava
42c6a147f2 extra argument removed 2019-04-01 11:30:36 -07:00
Swetha Mandava
bb0332437b fix readme 2019-03-28 16:10:54 -07:00
Przemek Strzelczyk
558917b3c2 Additional cleanup (BERT) 2019-03-27 17:47:46 +01:00
Przemek Strzelczyk
93fa95e3c7 Removing obsolete code (BERT) 2019-03-27 17:34:23 +01:00
Michał Krzysztof Filipiuk
041c35247b Fixed indentation 2019-03-25 12:52:08 -07:00
Swetha Mandava
dfbce5d178 removing multinode scripts 2019-03-21 23:02:48 -07:00
Przemek Strzelczyk
4b09b347a7 BERT README update 2019-03-19 17:12:43 +01:00
nvpstr
21024d9126
Merge pull request #15 from NVIDIA/nvpstr/master
Updating SSD and BERT.
2019-03-19 16:52:05 +01:00
Jonathan DEKHTIAR
f0ec1d4092
[UNet-TF Industrial] - Readme Fix 2019-03-19 16:36:25 +02:00
Przemek Strzelczyk
3d90b6dbf0 Updating SSD and BERT. 2019-03-19 15:16:07 +01: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
Nathan Luehr
134db41c81 Cleanup TF imagenet readme. 2018-05-03 08:47:20 -07:00
Nathan Luehr
2fdaec9c85 Add TensorFlow examples 2018-05-02 17:54:22 -07:00