Commit graph

111 commits

Author SHA1 Message Date
Przemek Strzelczyk 0663b67c1a Updating models 2019-07-08 22:51:28 +02:00
nvpstr f89dcca19d
Merge pull request #89 from GrzegorzKarchNV/readme-epochs
changed number of epochs in train scripts; removed number of epochs f…
2019-07-01 11:44:01 +02:00
nvpstr eabbedd1d4
Merge pull request #91 from GrzegorzKarchNV/readme-fix
fixed batches for mixed precision and fp32, added info when no input …
2019-07-01 11:43:29 +02:00
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