Commit graph

84 commits

Author SHA1 Message Date
Vinh Nguyen 7d3d1a028b
Include jupyter in this docker build 2019-06-05 18:38:36 +10: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
nvpstr 3454305d61
Merge pull request #58 from NVIDIA/nvpstr/master
Updating PyTorch models
2019-05-28 11:19:41 +02:00
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
nvpstr c71a3fe05b
Merge pull request #52 from sharathts/master
Fix README
2019-05-25 01:43:24 +02:00
nvpstr 3306ea2ab9
Merge branch 'master' into master 2019-05-25 01:43:11 +02:00
nvpstr 12c7bc2302
Merge pull request #45 from eric-haibin-lin/patch-1
Fix a few typos, and add gluonnlp implementation link
2019-05-25 01:37:31 +02:00
nvpstr 9712cf8281
Merge pull request #54 from vinhngx/master
adding resume capability for wget in case data downloads stop early
2019-05-25 01:37:07 +02:00
nvpstr 94472422fa
Merge pull request #55 from NVIDIA/nvpstr/master
Adding UNet(medical)/TF & updating other TF models
2019-05-25 01:36:46 +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
Vinh Nguyen 8662db35a7 fix readme: direct link for Apex. Fix download script location 2019-05-24 14:47:43 +10:00
Vinh Nguyen eee0490810 adding resume capability for wget in case downloads stop early 2019-05-24 14:41:23 +10:00
sharatht 604b64e55f resolve merge conflict readme 2019-05-23 12:29:46 -07:00
Sharath T S ac3e9ddb51
Merge branch 'master' into master 2019-05-23 12:26:15 -07:00
sharatht f2bc6ae8ab Fix README 2019-05-23 12:21:00 -07: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
Haibin Lin e46eed4c57
Update README.md 2019-05-18 22:41:42 -07: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