Commit graph

35 commits

Author SHA1 Message Date
Krzysztof Kudrynski 49e23b4597 Adding links to performance benchmark page 2021-07-21 14:39:48 +02:00
Lukasz Pierscieniewski 33110132cc [Convnets/TF] Basic CPU model support 2021-07-15 08:22:43 +00:00
Piotr Marcinkiewicz 96b70f5d16 [ResNet50/TF1][Triton] Bermuda 0.6.11 update 2021-07-12 10:19:44 +00:00
Lukasz Pierscieniewski 70e0d55e80 Add missing QAT training file 2021-06-10 10:20:15 +00:00
hXl3s 2d555548b6
[ResNet/TF] Fix gradient calculation for sync variable 2021-04-26 15:00:10 +02:00
kkudrynski 169b081827 [ConvNets/TF1] Added Triton for ResNet 2021-04-20 13:50:41 +02:00
Lukasz Pierscieniewski 43d061ae87 Fixed wrong shapes with BS=1 2021-03-11 10:31:06 +01:00
nv-kkudrynski 06298be8c3
Merge pull request #856 from hXl3s/convnets/docs_update
Document behavior when --num_iter < --warmup_steps
2021-03-05 12:48:33 +01:00
Lukasz Pierscieniewski 4f8aaa22b0 Document behaviour when --num_iter < --warmup_steps 2021-03-04 10:17:50 +01:00
Shashank Verma 3d0a6a8be8 Add TF Classification README
Signed-off-by: Shashank Verma <shashank3959@gmail.com>
2020-12-02 00:42:16 -08:00
Lukasz Pierscieniewski c9846ca618 Fix merge issues 2020-11-09 13:37:14 +01:00
kkudrynski b2e7f4a284 [ConvNets/TF] Performance fix 2020-11-03 14:54:54 +01:00
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