Commit graph

258 commits

Author SHA1 Message Date
Grzegorz Karch
d7c20e5bfc fixed batches for mixed precision and fp32, added info when no input is provided for inference, added units to expected training time 2019-06-27 01:48:46 -07:00
Grzegorz Karch
7eb5eece14 changed number of epochs in train scripts; removed number of epochs from readme for the audio samples 2019-06-26 08:57:23 -07:00
nvpstr
40e074257f
Merge pull request #65 from JanuszL/fix_RN50_pytorch_DALI
Fix typo in RN50 PyTorch readme, remove unneeded lines for DALI
2019-06-13 17:59:16 +02:00
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
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
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
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