Commit graph

258 commits

Author SHA1 Message Date
nv-kkudrynski 2ac7ceb853
Merge pull request #990 from ChenYuHo/patch-3
[NFC/TF] Set GPU index based on local_rank instead of rank
2021-11-04 14:54:04 +01:00
Lukasz Pierscieniewski dcd3bbac09 [Convnets/TF] TF-TRT support 2021-11-02 06:53:59 -07:00
Krzysztof Kudrynski a0c9442f63 Merge: [Convnets/TF] Move inference dependencies into separate container 2021-11-02 06:38:06 -07:00
Lukasz Pierscieniewski defaf144ba [Convnets/TF] Move inference dependencies into separate container 2021-11-02 06:38:05 -07:00
Jan Golda 43df08994f Removed pip upgrade from Dockerfiles 2021-10-28 16:59:30 +02:00
Sharath Turuvekere Sreenivas 26d8955cc5 [PyT/TF/TF2] remove install html2text requirement for NLP models 2021-10-08 01:30:31 -07:00
Lukasz Pierscieniewski 61bcab7a0e [Convnets/TF] Bumped onnx dependencies 2021-10-07 18:24:38 +00:00
Michal Marcinkiewicz 03a93a1cb2 [UNetmed/TF] Fix registered metric names 2021-09-21 05:00:26 -07:00
Tomasz Grel 88e5298421 [VAE-CF-TF] Modify result logging 2021-08-31 16:53:43 +00:00
Ahmad Kiswani 248927e6fd [SSD-TF1] moved cython to requirements.txt 2021-08-27 14:19:49 +00:00
kkudrynski b6e5ebdbc9 Typos and small fixes 2021-08-23 17:48:46 +02:00
Elton Chen-Yu Ho a5f26eeef7
Set GPU index based on local_rank instead of rank 2021-08-22 17:45:54 +08:00
Lukasz Pierscieniewski 69a26f5c76 [Multiple models] Added protocol specification to requirements.txt 2021-08-11 16:54:02 +00:00
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
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
Sharath TS 499fb1c5ad
[BERT/PyT] fix typos
* fix typos

* fix typo
2021-04-06 11:46:25 +02:00
Lukasz Pierscieniewski 43d061ae87 Fixed wrong shapes with BS=1 2021-03-11 10:31:06 +01:00
nv-kkudrynski 8e131dd43b
Merge pull request #862 from lazykyama/bugfix/update_data_downloader
Update URLs for GLUE downloader.
2021-03-09 17:06:45 +01:00
Kazuhiro Yamasaki c33299b322 Update URLs for GLUE downloader. 2021-03-07 13:27:22 +09: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
Abhishek Sawarkar edb16c5e12 pynvml added to Dockerfile 2021-03-03 13:59:13 -08:00
Sharath TS 04988752a8
[BERT/PyT][BERT/TF] Switch back to the original server for data download
* update - wiki download
2021-02-25 14:13:53 -08:00
Sharath TS 3459f97d58
[BERT/PyT][BERT/TF] Use mirror and wget (#833)
* Use mirror and wget

Mirror speedup download by 10x
wget fixes the stall with urllib

* add comment

* Update wikidownloader of bert tf1 to use mirror and wget

Co-authored-by: Swetha Mandava <sweth.mandava@gmail.com>
2021-02-17 14:49:39 -08:00
Ivana Jovicic 36c0549c58 updated Fine-tuning training performance table numbers for DGX A100
update has been made for Throughput- mixed precision number (8 GPUs)
updates.

-table columns for weak scaling were swapped. Fixed
2021-02-09 14:05:11 -08:00
Swetha Mandava 004385d0ce
Merge pull request #824 from swethmandava/master
Updating notebooks and gpu_affinity scripts for bert tf1
2021-02-03 11:09:03 -08:00
Swetha Mandava c0ed4be84a Updating notebooks and gpu_affinity scripts for bert tf1 2021-02-02 23:49:57 -08:00
nv-kkudrynski 3f9cdc6290
Merge pull request #772 from shashank3959/notebook-update
Update DLE READMEs
2021-01-08 23:48:27 +01:00
Swetha Mandava 92829376a1
Merge pull request #791 from swethmandava/master
Triton README
2020-12-18 16:22:33 -08:00
Swetha Mandava 3d6e3d612c fix incorrect link in triton readme 2020-12-18 16:19:58 -08:00
Swetha Mandava 99b1c898ce
Merge pull request #773 from swethmandava/master
removie trt, fix queuing delay typo in triton readme for bert
2020-12-03 08:24:48 -08:00
Swetha Mandava 33ea90e1cc removie trt, fix queuing delay typo in triton readme for bert 2020-12-03 08:23:07 -08:00
Shashank Verma 815da8fe49 Add TF Recommendation README
Signed-off-by: Shashank Verma <shashank3959@gmail.com>
2020-12-02 00:46:31 -08:00
Shashank Verma d045b0dba8 Add TF Translation README
Signed-off-by: Shashank Verma <shashank3959@gmail.com>
2020-12-02 00:46:01 -08:00
Shashank Verma 65053f844f Add TF Language Modeling README
Signed-off-by: Shashank Verma <shashank3959@gmail.com>
2020-12-02 00:45:02 -08:00
Shashank Verma 04cd41a6b0 Add TF Segmentation README 2020-12-02 00:43:57 -08:00
Shashank Verma cca2a54c43 Add TF Detection README
Signed-off-by: Shashank Verma <shashank3959@gmail.com>
2020-12-02 00:43:22 -08:00
Shashank Verma 3d0a6a8be8 Add TF Classification README
Signed-off-by: Shashank Verma <shashank3959@gmail.com>
2020-12-02 00:42:16 -08:00
kkudrynski 478d5657cd [WideAndDeep/TF] Update for 20.10 2020-11-26 17:42:52 +01:00
kkudrynski fa1ddc982e [WideAndDeep/TF] library version fix 2020-11-10 11:55:27 +01:00
Swetha Mandava 7e85343cdc
Merge pull request #749 from swethmandava/master
remove links to old ngc checkpoints for BERT TF
2020-11-09 09:14:00 -08:00
Swetha Mandava 1113674cca
remove links to old ngc checkpoints 2020-11-09 09:12:30 -08:00
Swetha Mandava 5ec39fc532
Merge pull request #739 from swethmandava/master
Perf number correction for BERT TF SQuAD Fp32
2020-11-09 06:50:58 -08:00
Swetha Mandava 002bcd8c45
update gluon version to with bert in readme 2020-11-09 06:50:24 -08:00
Lukasz Pierscieniewski c9846ca618 Fix merge issues 2020-11-09 13:37:14 +01:00