Commit graph

12 commits

Author SHA1 Message Date
Pablo Ribalta 6ca9637885 Improve the way profiling is performed
Signed-off-by: Pablo Ribalta <pribalta@nvidia.com>
2020-01-10 11:11:08 +01:00
Pablo Ribalta Lorenzo 51b910f42f Cleanup (#328)
Signed-off-by: Pablo Ribalta <pribalta@nvidia.com>
2019-12-03 09:52:59 +01:00
Przemek Strzelczyk b4aef9945b Adding VNet/TF 2019-12-02 15:57:25 +01:00
Przemek Strzelczyk 4e609f5555 New notebooks in UNet(industrial)/TF 2019-11-18 23:11:28 +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 882bc26d40 [UNet_Industrial/TF] Adding Jupyter notebooks and small fixes 2019-10-09 14:45:45 +02:00
nv-kkudrynski 66a447276e
Update readme with info about external ckpts (#244) 2019-10-04 15:56:15 +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
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
Jonathan DEKHTIAR f0ec1d4092
[UNet-TF Industrial] - Readme Fix 2019-03-19 16:36:25 +02: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