Commit graph

69 commits

Author SHA1 Message Date
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
Przemek Strzelczyk
d2bc3da0a1 Changes in TF models:
* added UNet for medical image segmentation
* added TF-AMP support for RN50
* small updates for other models (READMEs, benchmark & testing scripts)
2019-05-25 01:23:11 +02:00
nvpstr
2d33a72240
Merge pull request #48 from nv-kkudrynski/torchhub_exposure
hub exposure fix and improvements
2019-05-23 16:17:40 +02:00
Krzysztof Kudrynski
161a4ea165 caching of pre-trained weights added to entrypoints 2019-05-21 15:10:38 +02:00
Krzysztof Kudrynski
14fe91fad5 added missing nested-modules-discovery from global entrypoint like torch.hub 2019-05-21 13:32:36 +02:00
nvpstr
bc64ab1ac5
Merge pull request #46 from zhiqwang/master
Fix typo
2019-05-20 14:00:12 +02:00
nvpstr
3cf8d6c3c5
Merge pull request #47 from swethmandava/master
fix mismatch in cl arguments for BERT
2019-05-20 13:59:49 +02:00
Swetha Mandava
c5951be727
fix mismatch in cl arguments 2019-05-19 10:10:58 -07:00
zhiqiang
5ebe18f983 Fix typo 2019-05-19 18:45:53 +08:00
nvpstr
205058fd7c
Merge pull request #43 from nv-kkudrynski/torchhub_exposure
torch.hub entrypoints for tacotron2, waveglow and ncf
2019-05-15 18:26:14 +02:00
nvpstr
f191136579
Merge pull request #36 from khoa-ho/master
Add current models to README
2019-05-15 15:52:22 +02:00
nvpstr
c4a42e4acb
Merge pull request #42 from khoa-ho/patch-1
Typo/stylistic fix for BERT README title
2019-05-15 15:48:14 +02:00
Krzysztof Kudrynski
49acab486b torch.hub entrypoints for tacotron2, waveglow and ncf 2019-05-15 14:16:46 +02:00
Khoa Ho
b0d259264e
Typo/stylistic fix for README title 2019-05-14 10:30:27 -07:00
nvpstr
a9d0554e5d
Merge pull request #30 from swethmandava/smandava/readme_edit
changing perf numbers to steady state throughput
2019-05-13 17:32:30 +02:00
Khoa Ho
fcaa639bd9
Add current models to README 2019-05-03 15:29:25 -07:00
Swetha Mandava
5551752822
Update readme error 2019-04-18 11:51:46 -07:00
Swetha Mandava
83fceae509 changing perf numbers to steady state throughput 2019-04-17 14:16:40 -07:00
nvpstr
531a570c5c
Merge pull request #25 from skierat/patch-1
Update Dockerfile
2019-04-08 18:19:11 +02:00
nvpstr
51d3dceb3f
Merge pull request #24 from swethmandava/smandava/readme_edit
fix readme
2019-04-08 18:18:39 +02:00
Swetha Mandava
ec9eaf732a removing fast_math, etc from README 2019-04-04 13:14:01 -07:00
Swetha Mandava
42c6a147f2 extra argument removed 2019-04-01 11:30:36 -07:00
skierat
2409559c12
Update Dockerfile 2019-04-01 14:34:29 +02:00
Swetha Mandava
bb0332437b fix readme 2019-03-28 16:10:54 -07:00
nvpstr
2acfcb263a
Removing obsolete code (BERT)
Nvpstr/fix
2019-03-27 17:52:09 +01: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
Michał Krzysztof Filipiuk
041c35247b Fixed indentation 2019-03-25 12:52:08 -07:00
nvpstr
783f5facde
Merge pull request #19 from swethmandava/master
removing multinode scripts
2019-03-22 16:47:21 +01:00
Swetha Mandava
dfbce5d178 removing multinode scripts 2019-03-21 23:02:48 -07:00
nvpstr
0e66c6dabb
Merge pull request #16 from NVIDIA/nvpstr/master
BERT README update
2019-03-19 17:14:07 +01:00
Przemek Strzelczyk
4b09b347a7 BERT README update 2019-03-19 17:12:43 +01:00
nvpstr
21024d9126
Merge pull request #15 from NVIDIA/nvpstr/master
Updating SSD and BERT.
2019-03-19 16:52:05 +01:00
nvpstr
a68a2a0e72
Merge pull request #14 from DEKHTIARJonathan/patch-1
[UNet-TF  Industrial] - Readme Fix
2019-03-19 16:50:20 +01:00
Jonathan DEKHTIAR
f0ec1d4092
[UNet-TF Industrial] - Readme Fix 2019-03-19 16:36:25 +02:00
Przemek Strzelczyk
3d90b6dbf0 Updating SSD and BERT. 2019-03-19 15:16:07 +01:00
nvpstr
8a3d34e8a4
Merge pull request #13 from sharath-ts/master
Fix README and Dockerfile
2019-03-19 11:02:28 +01:00
sharatht
f1661d8ca4 Fix README and Dockerfile 2019-03-18 18:06:48 -07:00
nvpstr
b1ae8dd47c
Merge pull request #12 from NVIDIA/nvpstr/master
Adding 9 new models (6 in TensorFlow and 3 in PyTorch)
2019-03-18 20:57:20 +01: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
nvpstr
713d95411a
Merge pull request #11 from tgrel/master
Update NCF-PyTorch README
2019-03-15 17:31:47 +01:00
Tomasz Grel
7c10b43d3e update NCF README 2019-02-21 14:54:47 +01:00
Szymon Migacz
b050963444 GNMT update 2019-02-14 12:40:30 +01:00
Szymon Migacz
8efbea403f Squashed 'PyTorch/Translation/GNMT/' changes from 4dc145a..51a90b1
51a90b1 Feb 14, 2019 update

git-subtree-dir: PyTorch/Translation/GNMT
git-subtree-split: 51a90b1667e7c1d45bd68bf719f8bf5d4c4521e3
2019-02-14 12:40:30 +01:00
Szymon Migacz
b99d45b3f5 Revert "Updating GNMT"
This reverts commit 072523d5b2.
2019-02-14 12:39:20 +01:00
nvpstr
072523d5b2 Updating GNMT 2019-01-25 16:20:05 +01:00
nvpstr
a1aff31c42 Adding training improvements. 2019-01-25 16:02:48 +01:00
Przemek Strzelczyk
a8e1d1cd83 Adding ResNet50v1.5 to MxNet/Classification 2019-01-23 17:14:51 +01:00
Przemek Strzelczyk
edc3486e6f Small fix to README 2019-01-23 17:08:35 +01:00
Przemek Strzelczyk
d755a84080 Adding Transformer to PyTorch/Translation 2019-01-23 17:03:45 +01:00