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 |
|
|
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 |
|