Commit graph

410 commits

Author SHA1 Message Date
nvpstr
34a01f7b72
Merge pull request #77 from GrzegorzKarchNV/warmup
updated readme: text.txt->phrase.txt, updated inference results
2019-06-13 17:35:33 +02:00
nvpstr
9df04c29c9
Merge pull request #79 from nv-kkudrynski/ssd_torchhub_exposure
ssd exposure via hubconf.py (together with ssd utils)
2019-06-13 17:35:16 +02:00
Krzysztof Kudrynski
a8328ce169 ssd exposure via hubconf.py (together with ssd utils) 2019-06-12 17:41:57 +02:00
Grzegorz Karch
cd284a62ec changed text to phrase as input to inference in the readme 2019-06-07 17:37:30 +02:00
Grzegorz Karch
dc7cd1ae5b updated readme 2019-06-07 17:37:30 +02:00
gkarch
79b9a48bf1 added warmup parameter to tacotron2 inference; updated README 2019-06-07 17:37:30 +02:00
nvpstr
4aa6f17167
Merge pull request #76 from GrzegorzKarchNV/torchhub
added text_to_sequence in the hubconf model
2019-06-07 15:11:28 +02:00
gkarch
362dfe6b3b added text_to_sequnce in the hubconf model 2019-06-07 14:52:23 +02:00
nvpstr
da165c3a52
Merge pull request #73 from GrzegorzKarchNV/master
updated readme - warmup for inference, updated results for 19.05-py3 container
2019-06-06 16:03:19 +02:00
nvpstr
bf1b29d73e
Merge pull request #63 from khcs/bert/pretrain-fix
Fix FLAG.save_checkpoint_steps naming inconsistency.
2019-06-06 16:02:28 +02:00
nvpstr
698e775680
Merge pull request #72 from jbaczek/master
Add missing submodule for PyT Transformer
2019-06-06 16:01:51 +02:00
nvpstr
c5bbd74f87
Merge pull request #62 from NVIDIA/nvpstr-patch-1
Update README.md
2019-06-06 15:58:28 +02:00
gkarch
2f34442146 update perf numbers 2019-06-06 15:27:21 +02:00
gkarch
4dd9ecfcfa updated readme - warmup for inference 2019-06-06 14:16:51 +02:00
jbaczek
2774e849be remove empty dir 2019-06-05 16:20:30 +02:00
jbaczek
743162906c fix submodule path 2019-06-05 16:17:24 +02:00
jbaczek
698516f8f7 add missing submodule 2019-06-05 16:14:41 +02:00
nvpstr
2619f172c7
Merge pull request #67 from GrzegorzKarchNV/master
Update models.py - fix fp16 inference
2019-06-05 14:45:27 +02:00
Vinh Nguyen
7d3d1a028b
Include jupyter in this docker build 2019-06-05 18:38:36 +10:00
Vinh Nguyen
1cac44e014 add this repo to docker build 2019-06-05 16:32:58 +10:00
GrzegorzKarchNV
867ab68fad
Update models.py 2019-06-04 15:44:05 +02:00
Janusz Lisiecki
20f09772f8 Fix typo in RN50 PyTorch readme, remove unneeded lines for DALI
Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
2019-06-03 12:03:18 +02:00
Hoo Chang Shin
37c4083c4b change to BERT-base, and checkpoints-dir. 2019-05-31 10:31:26 -04:00
Hoo Chang Shin
1d8ac4b295 Merge branch 'bert/pretrain-fix' 2019-05-30 12:58:03 -04:00
Hoo Chang Shin
65bf7dbe48 initial commit on training PubMed-gvocab-base. 2019-05-30 12:55:57 -04:00
Hoo Chang Shin
f3c1cc7204 Fix FLAG.save_checkpoint_steps naming inconsistency. 2019-05-30 11:37:46 -04:00
nvpstr
e5a0a056d2
Update README.md 2019-05-30 13:42:31 +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
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