Commit graph

23 commits

Author SHA1 Message Date
kkudrynski 5d36b4fd2f Fixing hyperlinks 2020-09-08 16:10:32 +02:00
Łukasz Pierścieniewski 8bd6dd14d3 Document synthetic dataset options 2020-08-20 16:21:50 +02:00
Dheeraj Peri 4cb58b61c7 Add licenses to new files 2020-07-06 12:01:17 -07:00
Dheeraj Peri 57d9cc0444 Update instructions 2020-07-01 13:19:26 -07:00
Dheeraj Peri 00bde50081 Remove folder for QAT 2020-07-01 13:16:58 -07:00
Dheeraj Peri ed7831daa0 Update instructions and output node name 2020-07-01 13:09:39 -07:00
Dheeraj Peri da2e33a67d Merge branch 'master' into rn50_qat_v2 2020-07-01 11:50:10 -07:00
Dheeraj Peri 6d2357a9b8 code Refactor 2020-07-01 11:42:19 -07:00
Dheeraj Peri e0f399def4 Update frozen graph script and instructions 2020-06-30 11:49:00 -07:00
Przemek Strzelczyk 4eaa4434de [ConvNets/TF] Adding support for Ampere 2020-06-27 09:24:41 +02:00
Dheeraj Peri c4f90be499 Add QAT instructions for RN50 2020-06-21 16:41:23 -07:00
Krzysztof Kudrynski f11884b38a minor readme updates 2020-06-12 13:50:44 +02:00
Przemek Strzelczyk fed7ba99cd [ConvNets/TF] Updating RN50, Adding ResNext and SE-ResNext 2020-06-12 12:38:25 +02:00
Przemek Strzelczyk 11f6038cd8 Updating RN50/TF 2019-10-21 19:49:00 +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
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 e5a0a056d2
Update README.md 2019-05-30 13:42: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
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