Commit graph

21 commits

Author SHA1 Message Date
Swetha Mandava
b82c372047 triton v2 api, download mrpc fix, update for mpi 4.2 2020-08-10 11:09:40 -07:00
Swetha Mandava
c2051b49eb pulling v1 tritonserver 2020-07-20 13:23:04 -07:00
Krzysztof Kudrynski
878b004e6b [NCF/TF, WideAndDeep/TF] Updating for Ampere, [BERT/TF] MRPC support 2020-07-14 12:34:25 +02:00
Przemek Strzelczyk
96138d5087 [BERT/TF] Updating for Ampere 2020-07-04 01:00:48 +02:00
Swetha Mandava
942b09611c change default dllog path, disable horovod for 1 gpu 2020-05-14 14:25:37 -07:00
Swetha Mandava
398dc781a1 amp env variable to amp api 2020-05-12 22:18:05 -07:00
Przemek Strzelczyk
cef4bab3fc Adding BERT/TF Inference with TensorRT; minor README fixes 2020-02-05 14:21:13 +01:00
Swetha Mandava
23d56b7cee replacing / with - in init_checkpoint string for printing 2019-12-13 19:00:40 -08:00
Swetha Mandava
2a93a1efa9 mkdir missing 2019-11-18 18:49:02 -08:00
Przemek Strzelczyk
9cd3946603 Updating BERT/TF
- Pre-training and Finetuning on BioMedical tasks and corpus
- Disabling Grappler Optimizations for improved performance
2019-11-04 23:18:08 +01:00
Przemek Strzelczyk
a98df279fe [BERT/TF] Added multi-node support 2019-09-13 19:12:50 +02:00
Przemek Strzelczyk
fa400a7367 Adding Jasper/PyT 2019-07-26 20:08:16 +02:00
Przemek Strzelczyk
15075aab0c Removing obsolete scripts 2019-07-25 20:15:52 +02:00
Przemek Strzelczyk
8218872051 Updating BERT with TRT-IS support and new results 2019-07-25 16:53:05 +02:00
Przemek Strzelczyk
0663b67c1a Updating models 2019-07-08 22:51:28 +02:00
nvpstr
b7eff3da41
Merge pull request #60 from swethmandava/smandava/bug_fix
num_gpus pretraining fix
2019-05-30 13:25:58 +02:00
Swetha Mandava
b25181c9b8 num_gpus pretraining fix 2019-05-28 09:52:52 -07:00
Przemek Strzelczyk
558917b3c2 Additional cleanup (BERT) 2019-03-27 17:47:46 +01:00
Przemek Strzelczyk
93fa95e3c7 Removing obsolete code (BERT) 2019-03-27 17:34:23 +01:00
Swetha Mandava
dfbce5d178 removing multinode scripts 2019-03-21 23:02:48 -07: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