Pablo Ribalta
70f3e4362c
Remove dllogger and fix bugs from GH
...
Signed-off-by: Pablo Ribalta <pribalta@nvidia.com>
2020-04-08 09:40:52 +02:00
Pablo Ribalta Lorenzo
e9fc30acb8
Fix requirements
...
Signed-off-by: Pablo Ribalta Lorenzo <pribalta@nvidia.com>
2020-04-02 07:16:47 +00:00
PrzemekS
c6ebe1c756
Merge pull request #377 from NVIDIA/vnet_benchmark_fix
...
Enable rewrite for inference
2020-03-09 17:34:03 +01:00
Przemek Strzelczyk
72ef5f45cb
[MaskRCNN/TF2] Adding MaskRCNN for TF1 and TF2
2020-03-05 09:49:01 +01:00
Przemek Strzelczyk
dee0fe36c5
[UNet(med)/TF] New perf results, cross-validation, optimizer and README
2020-03-02 14:29:12 +01:00
Pablo Ribalta
912c394db3
typo
...
Signed-off-by: Pablo Ribalta <pribalta@nvidia.com>
2020-01-22 22:26:10 +01:00
Pablo Ribalta
171fa76925
Enable rewrite for inference
...
Signed-off-by: Pablo Ribalta <pribalta@nvidia.com>
2020-01-20 11:27:33 +01:00
Pablo Ribalta
8a863783dc
Cosmetics
...
Signed-off-by: Pablo Ribalta <pribalta@nvidia.com>
2020-01-10 11:17:33 +01:00
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