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

31 commits

Author SHA1 Message Date
Pablo Ribalta Lorenzo 5e7dd3b91b
[UNet_Ind/TF] Fix number of gpus 
Signed-off-by: Pablo Ribalta Lorenzo <pribalta@nvidia.com>
 2021-04-07 14:45:56 +02:00
Shashank Verma 04cd41a6b0 Add TF Segmentation README 2020-12-02 00:43:57 -08:00
Pablo Ribalta Lorenzo 81e86366f3 Add --gpus flag to docker run 
Signed-off-by: Pablo Ribalta Lorenzo <pribalta@nvidia.com>
 2020-11-02 14:31:31 +01:00
kkudrynski 437b950d5b Fixed links in readme 2020-09-14 23:58:38 +02:00
kkudrynski 5d36b4fd2f Fixing hyperlinks 2020-09-08 16:10:32 +02:00
Pablo Ribalta Lorenzo fb40734b31
Remove autobench scripts (#647) 
Signed-off-by: Pablo Ribalta <pribalta@nvidia.com>
 2020-08-12 11:48:46 -07:00
Michał Marcinkiewicz adddf66b9c
Update README.md 2020-08-02 19:35:09 +02:00
Przemek Strzelczyk 79d4ced0be Adding 3DUnet/TF 2020-07-04 03:28:33 +02:00
Przemek Strzelczyk 9f4678de1b [UNet industrial/TF] Updating for Ampere 2020-07-04 02:17:58 +02:00
Przemek Strzelczyk b27abeba07 [UNet_medical/TF1&2] Updating for Ampere 2020-07-04 01:42:09 +02:00
Przemek Strzelczyk 76a056cd33 [VNet/TF] Updating for 20.06 container 2020-07-04 01:37:11 +02:00
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