Commit graph

225 commits

Author SHA1 Message Date
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
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
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
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
Krzysztof Kudrynski 14fe91fad5 added missing nested-modules-discovery from global entrypoint like torch.hub 2019-05-21 13:32:36 +02:00
Krzysztof Kudrynski 49acab486b torch.hub entrypoints for tacotron2, waveglow and ncf 2019-05-15 14:16:46 +02:00
skierat 2409559c12
Update Dockerfile 2019-04-01 14:34:29 +02:00
sharatht f1661d8ca4 Fix README and Dockerfile 2019-03-18 18:06:48 -07:00
Przemek Strzelczyk 7a8544e0a2 Adding 9 new models: ResNet50/TF, SSD/TF, BERT/TF, NCF/TF, UNet/TF, GNMT/TF, SSD/PyT, MaskRCNN/PyT, Tacotron2+Waveglow/PyT 2019-03-18 19:45:05 +01:00
Tomasz Grel 7c10b43d3e update NCF README 2019-02-21 14:54:47 +01:00
Szymon Migacz b050963444 GNMT update 2019-02-14 12:40:30 +01:00
Szymon Migacz b99d45b3f5 Revert "Updating GNMT"
This reverts commit 072523d5b2.
2019-02-14 12:39:20 +01:00
nvpstr 072523d5b2 Updating GNMT 2019-01-25 16:20:05 +01:00
nvpstr a1aff31c42 Adding training improvements. 2019-01-25 16:02:48 +01:00
Przemek Strzelczyk edc3486e6f Small fix to README 2019-01-23 17:08:35 +01:00
Przemek Strzelczyk d755a84080 Adding Transformer to PyTorch/Translation 2019-01-23 17:03:45 +01:00
Przemek Strzelczyk 585973766f Adding NCF to PyTorch/Recommendation 2019-01-23 16:59:07 +01:00
nvpstr 3ac7dc4957 Adding ResNet50 v1.5. 2018-10-01 22:51:24 +02:00
Szymon Migacz 7d69e136a3 Added PyTorch GNMT implementation 2018-08-07 16:27:43 +02:00
Christian Sarofeen 76a3d10a19 Add pytorch examples 2018-05-02 18:33:56 -07:00