Commit graph

27 commits

Author SHA1 Message Date
Andrzej Sulecki ab6267ad33 [internal/CI][PyT/ConvNets] Move to launcher 2021-07-26 21:18:30 +00:00
Krzysztof Kudrynski 49e23b4597 Adding links to performance benchmark page 2021-07-21 14:39:48 +02:00
Yuanzhe Dong 3d8d878489
[ConvNets/PyT] update triton repo url in readmes 2021-07-21 12:53:10 +02:00
Andrzej Sulecki a1bbe6687e [PyT/ConvNets] Fixing bug in model2onnx 2021-07-12 10:20:43 +00:00
Andrzej Sulecki c481324031 [PyT/EfficientNet] Update README 2021-07-01 15:11:18 +02:00
Piotr Marcinkiewicz 5c33a8289b
[ResNet50/PyT] Triton perf fix
* ResNet50/PyT Triton ONNXruntime fix with env flag

Scripts were modified to fix missing ORT_TENSORRT_FP16_ENABLE flag for
Triton Inference Server with ONNXRuntime and TensorRT execution provider.

* ResNet50/PyT TensorRT FP16 support fixed

ONNX to TensorRT converter was fixed to force FP16 precision for
TensorRT networks.
2021-06-16 16:04:22 +02:00
Andrzej Sulecki 555b84b3b1 [ConvNets/PyT] Adding checkpoints for EfficientNet/PyT, Squeeze&Excitation can use Conv or Linear layer depending on --trt switch. 2021-05-26 11:03:13 +00:00
kkudrynski 7bdfc81d25 [ConvNets/PyT] Triton and performance numbers updated 2021-04-20 12:54:18 +02:00
kkudrynski 7fbe4ab64c [ConvNets/PyT] QAT for EfficientNet 2021-04-13 19:05:27 +02:00
kkudrynski 2bdf2775e3 [ConvNets/PyT] EfficientNet release 2021-04-09 23:12:57 +02:00
Andrzej Sulecki b946389828 Fix missing multiproc in example training script 2021-03-22 11:42:01 +01:00
kkudrynski 097c7ff917 [ConvNets/PyT] Native AMP and update to 20.12 2021-03-04 16:14:35 +01:00
Shashank Verma acc0d2a7fa Add image classification README
Content obtained from NGC Collections

Signed-off-by: Shashank Verma <shashank3959@gmail.com>
2020-12-02 00:19:03 -08:00
Lukasz Pierscieniewski c9846ca618 Fix merge issues 2020-11-09 13:37:14 +01:00
kkudrynski 9061083098 [ConvNets/Pyt] Pretrained weights usage guidelines 2020-10-21 17:19:07 +02:00
kkudrynski f217ab11a4 [COnvNets/Pyt] Triton Deployment 2020-10-14 20:43:50 +02:00
Łukasz Pierścieniewski 72f40b825a Fixed distributed checkpoint loading 2020-09-18 12:28:49 +02:00
Przemek Strzelczyk 46ff3707e0 [ConvNets/PyT] Adding support for Ampere and 20.06 container 2020-06-27 09:32:20 +02:00
Elton Chen-Yu Ho accf26d1d8
Fix a typo in PyTorch ConvNets README.md 2020-06-22 14:32:37 +08:00
Krzysztof Kudrynski f0ef8493eb ConvNets update 2019-12-20 14:54:58 +01:00
Przemek Strzelczyk 5562ab767a Adding SE-ResNext and ResNext / PyT 2019-12-15 05:13:59 +01: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
Janusz Lisiecki 20f09772f8 Fix typo in RN50 PyTorch readme, remove unneeded lines for DALI
Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
2019-06-03 12:03:18 +02:00
Przemek Strzelczyk 5eaebefcd0 Updating PyTorch models
* New training scripts & results for RN50 PyT
* Documentation updates for Trasformer, SSD and MaskRCNN
2019-05-27 15:24:14 +02:00
nvpstr a1aff31c42 Adding training improvements. 2019-01-25 16:02:48 +01:00
nvpstr 3ac7dc4957 Adding ResNet50 v1.5. 2018-10-01 22:51:24 +02:00