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