Commit graph

1008 commits

Author SHA1 Message Date
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
Adrian Lancucki 8d8c524df6
[FastPitch/PyT] Fix updated regulate_len 2021-05-21 11:35:47 +02:00
Grzegorz Karch 5a8521ee05 [Tacotron2/PyT] rename common->tacotron2_common 2021-05-18 16:28:11 +00:00
Jan Golda 5afd63dde0 [MaskRCNN/TF2] Update 2021-05-17 09:22:33 +02:00
Michal Futrega 905e9e507e [nnUnet/PyT] Add support for Triton 2021-05-14 15:41:09 +02:00
nv-kkudrynski 2a2735fed1
Merge pull request #935 from alancucki/readme-update
[FastPitch/PyT] Update model description
2021-05-14 14:09:58 +02:00
Adrian Lancucki 0c04f3caaf Update model description 2021-05-14 14:05:32 +02:00
nv-kkudrynski 97942685fb
Merge pull request #934 from PiotrBigajNV/master
[Wide&Deep/TF2] Add Dockerfile symlink to W&D
2021-05-10 12:01:17 +02:00
Piotr Bigaj ceaa6c63bd Add Dockerfile symlink to W&D 2021-05-10 11:39:14 +02:00
Robert Clark 1982f5bacd
[BERT/PyT] Update GLUE Download Tool
The linked GLUE downloader script has several issues which prevent it
from downloading the MRPC components. Creating a new fork of the linked
gist allows these items to be sorted out.

Signed-Off-By: Robert Clark <roclark@nvidia.com>
2021-04-30 16:43:30 -05:00
hXl3s 2d555548b6
[ResNet/TF] Fix gradient calculation for sync variable 2021-04-26 15:00:10 +02:00
nv-kkudrynski 01201316f8
Merge pull request #924 from NVIDIA/gh/release
[MaskRCNN/TF2] Readme fix
2021-04-24 17:55:58 +02:00
kkudrynski e438e68236 [MaskRCNN/TF2] Readme fix 2021-04-24 17:54:33 +02:00
Subhankar Ghosh ec2bef8444
[EfficientNet/TF2] Fix inference numbers and calculations
Co-authored-by: Subhankar Ghosh <subhankarg@nvidia.com>
2021-04-22 13:52:13 +02:00
nv-kkudrynski da18e57a91
BERT/TF2 added to main readme 2021-04-21 14:34:48 +02:00
nv-kkudrynski 5b1dddc696
Merge pull request #921 from NVIDIA/gh/release
[BERT/TF2] Release
2021-04-21 14:19:44 +02:00
kkudrynski afb7dd105b [BERT/TF2] Release 2021-04-21 14:16:30 +02:00
nv-kkudrynski bd257e1494
Merge pull request #920 from NVIDIA/gh/release
Gh/release
2021-04-20 13:54:27 +02:00
kkudrynski 169b081827 [ConvNets/TF1] Added Triton for ResNet 2021-04-20 13:50:41 +02:00
kkudrynski 7bdfc81d25 [ConvNets/PyT] Triton and performance numbers updated 2021-04-20 12:54:18 +02:00
byshiue 5b9787c4a0
Merge pull request #917 from NVIDIA/byshiue-patch-1
[FT] Remove the source codes of FT and add the link of new repo
2021-04-17 11:05:40 +08:00
bhsueh fd13bbdad1 [FT] Remove the source codes of FT and add the link of new repo 2021-04-16 20:03:43 -07:00
bhsueh d58ccd79e5 Merge branch 'master' into byshiue-patch-1 2021-04-16 19:59:09 -07:00
nv-kkudrynski 9f8b7de239
Merge pull request #916 from alancucki/Jasper/pyyaml
[Jasper/PyT] Update PyYAML
2021-04-15 21:55:00 +02:00
Adrian Lancucki d7df37ab41 Update PyYAML due to security vulnerability 2021-04-15 15:00:57 +02:00
nv-kkudrynski 2b20ca80cf
PTL link fix 2021-04-14 19:37:40 +02:00
nv-kkudrynski 26206dbf87
Merge pull request #914 from NVIDIA/gh/release
[ConvNets/PyT] QAT for EfficientNet
2021-04-13 19:15:50 +02:00
kkudrynski 7fbe4ab64c [ConvNets/PyT] QAT for EfficientNet 2021-04-13 19:05:27 +02:00
nv-kkudrynski 93ee37eeb9
Merge pull request #912 from subhankar-ghosh/master
Update Computer vision model list
2021-04-13 16:22:09 +02:00
nv-kkudrynski 9db23472d3
Merge pull request #913 from NVIDIA/gh/release
[FastPitch/PyT] Added Triton
2021-04-13 16:21:40 +02:00
kkudrynski afe00719ca [FastPitch/PyT] Added Triton 2021-04-13 16:18:11 +02:00
Subhankar Ghosh 09dbddc365 Update Computer vision model list 2021-04-13 02:03:54 -07:00
Subhankar Ghosh 529db244cd
[EfficientNet/TF2] Update training time typo 2021-04-10 22:11:17 +02:00
nv-kkudrynski 82422354a9
Merge pull request #908 from NVIDIA/gh/release
EfficientNets
2021-04-09 23:36:21 +02:00
kkudrynski 4a66a008c4 [ConvNets/TF2] EfficientNet release 2021-04-09 23:32:53 +02:00
kkudrynski 2bdf2775e3 [ConvNets/PyT] EfficientNet release 2021-04-09 23:12:57 +02:00
nv-kkudrynski 402fcbc323
Merge pull request #906 from NVIDIA/gh/release
[Resnet/MX] Update
2021-04-07 17:50:56 +02:00
kkudrynski 8c42220b49 [Resnet/MX] Update 2021-04-07 17:46:50 +02:00
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
nv-kkudrynski 4922b484ae
Merge pull request #904 from PiotrBigajNV/master
Change evaluation performance into inference performance for Wide And Deep TF2 model
2021-04-07 11:26:02 +02:00
Piotr Bigaj f9cc3713ac Change evaluation performance into inference performance 2021-04-07 10:39:53 +02:00
nv-kkudrynski 0a3a4d0916
Merge pull request #901 from szmigacz/txl-update3
[TXL/PyT] Update for PyT Transformer-XL:
2021-04-06 12:00:46 +02:00
Sharath TS 499fb1c5ad
[BERT/PyT] fix typos
* fix typos

* fix typo
2021-04-06 11:46:25 +02:00
nv-kkudrynski a71bae7cda
Merge pull request #893 from NVIDIA/pribalta/fixes_to_nnunet_readme
nnUNet: Add reference to PyTorch Lightning to readme
2021-04-06 11:40:28 +02:00
byshiue a64351f3f1
[FT] Add the deprecation in readme 2021-04-05 15:49:19 +08:00
Szymon Migacz 09a17a0f33 [TXL/PyT] Update for PyT Transformer-XL:
* WAR for issues with logging in distributed setting
* improved CPU-GPU affinity
* minor optimizations for the model
2021-04-02 22:07:48 -07:00
Pablo Ribalta Lorenzo 86ae47f0db Add PTL to the README
Signed-off-by: Pablo Ribalta Lorenzo <pribalta@nvidia.com>
2021-03-29 16:41:02 +02:00
nv-kkudrynski 9750daf189
Merge pull request #890 from alancucki/jasper/update_readme
Update README
2021-03-26 11:20:49 +01:00
Adrian Lancucki aebcac25a8 Fix README typos 2021-03-26 10:25:57 +01:00
nv-kkudrynski d47ce467ad
Updated model list 2021-03-25 18:54:37 +01:00