Commit graph

18 commits

Author SHA1 Message Date
Ahmad Kiswani 0b3f770b0a [SSD/PyT] New release vs 21.05 container 2021-11-09 03:16:21 -08:00
Andrei Shumak 58df33d1a2 [SSD/PyT] Fixes 2021-08-30 12:47:32 -07:00
Andrei Shumak d6cd6b869c [SSD/PyT] Improved logging 2021-08-27 07:16:50 -07:00
Krzysztof Kudrynski 49e23b4597 Adding links to performance benchmark page 2021-07-21 14:39:48 +02:00
kkudrynski 81ee705868 [PyTorchHub] Adding licences to entrypoint scripts 2021-06-18 14:56:07 +00:00
Jan Golda 778583481b [TorchHub] restructured hubconf and updated SSD and Tacotron2/WaveGlow entrypoints 2021-06-07 12:20:51 +00:00
kkudrynski 437b950d5b Fixed links in readme 2020-09-14 23:58:38 +02:00
Krzysztof Kudrynski 33bdf65b18 readme fixes 2020-07-08 12:55:56 +02:00
Krzysztof Kudrynski 31ca062d93 [SSD/PyT] Updating for Ampere 2020-07-07 23:41:00 +02:00
skierat 2fb7c8f472
Merge pull request #149 from HanbumKo/HanbumKo-patch-1
Update inference.py
2019-12-09 11:32:36 +01:00
Przemek Strzelczyk e470c2150a Updating RN50/MxNet 2019-10-21 19:20:40 +02:00
Przemek Strzelczyk 4cce4d88e6 Updating SSD/PyT 2019-09-12 14:33:49 +02:00
HanbumKo 8d365d4b1f
Update inference.py
(mean, std) isn't used in load_image but normalize.
2019-08-06 14:30:04 -07: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 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
skierat 2409559c12
Update Dockerfile 2019-04-01 14:34:29 +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