Commit Graph

1008 Commits

Author SHA1 Message Date
Andrzej Sulecki 4f2c6922bd [ConvNets/PyT] TorchScriptable ConvNets 2021-11-09 13:42:18 -08:00
Dawid Majchrowski 3d3250a3ae [TFT/PyT] Add inference section to README and update Dockerfile 2021-11-09 07:41:03 -08:00
Krzysztof Kudrynski 46888c8296 Merge: [ConvNets/PyT] Exposing to PytorchHub 2021-11-09 05:18:24 -08:00
Krzysztof Kudrynski 83e057630c [ConvNets/PyT] Exposing to PytorchHub 2021-11-09 05:18:23 -08:00
Krzysztof Kudrynski 7ebda8e181 Merge: [TSPP/PyT][TFT/PyT] README fixes and removing redundancies 2021-11-09 03:23:13 -08:00
Kyle Kranen f1120f433f [TSPP/PyT][TFT/PyT] README fixes and removing redundancies 2021-11-09 03:23:12 -08:00
Ahmad Kiswani 0b3f770b0a [SSD/PyT] New release vs 21.05 container 2021-11-09 03:16:21 -08:00
Krzysztof Kudrynski ffad84899b Merge: [MRCNN/PyT] Update opencv-python version, fix numpy version 2021-11-08 14:25:16 -08:00
Krzysztof Kudrynski 0b7816ea4d Merge: [Tacotron2/PyT] Using native amp 2021-11-08 14:23:51 -08:00
Grzegorz Karch 4fe3b4e4ae [Tacotron2/PyT] Using native amp 2021-11-08 14:23:50 -08:00
Krzysztof Kudrynski 4a54f41128 Merge: [WIDENDEEP/TF2] Update NVTabular version and add various model optimizations 2021-11-08 14:10:12 -08:00
Dawid Majchrowski 01a9f5b48c [WIDENDEEP/TF2] Update NVTabular version and add various model optimizations 2021-11-08 14:10:12 -08:00
Kyle Kranen 38d72e07d2 [TSPP/PyT][TFT/PyT] Initial release 2021-11-08 14:08:58 -08:00
Andrei Shumak 8f86776757 Merge branch 'asulecki/dllogger_install_fix' into 'internal/main'
Fix DLLogger install

See merge request dl/JoC/examples!367
2021-11-08 06:36:33 -08:00
nv-kkudrynski 8ed4627567
Merge pull request #1018 from Peter-1213/master
[Transformer/PyT] Fix a logic flaw that causes checkpoints to be saved at wrong epochs in the transformer example.
2021-11-08 12:27:49 +01:00
Shriya Balaji Palsamudram 53f95e5a5f [MRCNN-PYT] Update opencv-python version, fix numpy version 2021-11-04 14:13:48 -07:00
nv-kkudrynski 2592d5a02c
Merge pull request #931 from roclark/update-glue-downloader
[BERT/PyT] Update GLUE Download Tool
2021-11-04 15:08:15 +01:00
nv-kkudrynski 2ac7ceb853
Merge pull request #990 from ChenYuHo/patch-3
[NFC/TF] Set GPU index based on local_rank instead of rank
2021-11-04 14:54:04 +01:00
asulecki cab1e33745 Fix DLLogger install 2021-11-02 15:12:02 +01:00
Alexandre Milesi 22d6621dcd [SE3Transformer/DGLPyT] Better low memory mode 2021-11-02 15:06:21 +01:00
Lukasz Pierscieniewski dcd3bbac09 [Convnets/TF] TF-TRT support 2021-11-02 06:53:59 -07:00
Krzysztof Kudrynski a0c9442f63 Merge: [Convnets/TF] Move inference dependencies into separate container 2021-11-02 06:38:06 -07:00
Lukasz Pierscieniewski defaf144ba [Convnets/TF] Move inference dependencies into separate container 2021-11-02 06:38:05 -07:00
Jan Golda 43df08994f Removed pip upgrade from Dockerfiles 2021-10-28 16:59:30 +02:00
Peter-1213 9cf6c3824b Fix a logic flaw that causes checkpoints to be saved at wrong epochs. 2021-10-27 01:22:33 +08:00
Shriya Balaji Palsamudram abe062867f [MRCNN] use native AMP, upgrade opencv-python 2021-10-26 17:00:32 +00:00
Andrei Shumak cc194960e6 Merge branch 'alexandrem/se3-a100-tests' into 'internal/main'
[internal/CI] Add SE3T A100 tests

See merge request dl/JoC/examples!348
2021-10-21 00:11:31 -07:00
Alexandre Milesi 67c7afa053 [internal/CI] Add SE3T A100 tests 2021-10-21 00:11:30 -07:00
Alexandre Milesi 9b86d8c0b8 [SE3Transformer/DGLPyT] Fix multiple shapes mismatch with specific hyperparams 2021-10-20 10:16:49 -07:00
Michal Futrega 028534a5b9 [nnUNet/PyT] Add Jupyter notebook with BraTS21 solution 2021-10-20 10:03:05 -07:00
Krzysztof Kudrynski 8c98f155e0 Merge branch 'akohlhoff/kaldi_release'
[Kaldi] Update to 21.08
2021-10-18 01:43:26 -07:00
nv-kkudrynski a0fa02f592
Merge pull request #1015 from szmigacz/gnmt-update2
[GNMT/PyT] requirements.txt: updated pytablewriter to 0.64.0
2021-10-18 10:31:05 +02:00
michalm 0db746b4ab [RoseTTAFold] Initial release 2021-10-15 15:46:41 +02:00
Szymon Migacz 11665f9cb2 [GNMT/PyT] requirements.txt: updated pytablewriter to 0.64.0 2021-10-13 09:40:51 -07:00
Alec Kohlhoff 1a5c7556b5 [Kaldi] Update to 21.08 2021-10-12 17:45:31 -07:00
Sharath Turuvekere Sreenivas 26d8955cc5 [PyT/TF/TF2] remove install html2text requirement for NLP models 2021-10-08 01:30:31 -07:00
Jan Golda a64b105b8c [ConvNets/PyT] Remove pytorch_quantization from requirements.txt since it is already in container 2021-10-07 18:34:19 +00:00
Adrian Lancucki 054fed043f [FastPitch/PyT] Drop parselmouth dependency 2021-10-07 11:26:31 -07:00
Lukasz Pierscieniewski 61bcab7a0e [Convnets/TF] Bumped onnx dependencies 2021-10-07 18:24:38 +00:00
Adrian Lancucki f24917b3ee Add a script to download model from NGC 2021-09-29 14:51:16 +00:00
Alexandre Milesi f4d6a00cb5 [SE3Transformer/PyT] Migrate to new location 2021-09-24 15:14:10 +02:00
Adrian Lancucki 6d8ddbbc67 Fix dithering bug in DALI pipeline 2021-09-22 12:20:54 +02:00
Michal Marcinkiewicz 03a93a1cb2 [UNetmed/TF] Fix registered metric names 2021-09-21 05:00:26 -07:00
Alexandre Milesi 8d31298c56 [SE3Transformer/PyT] Fix multi-GPU training bug 2021-09-14 13:04:04 +00:00
Mikolaj Blaz 649776f79a [QuartzNet/PyT] Release QuartzNet model 2021-09-14 06:03:36 -07:00
Lukasz Pierscieniewski 88eb3cff2f [RN50/MX] Skip first iterations for better metrics calculation 2021-09-08 08:09:23 -07:00
Adrian Lancucki 6d762dacdc rm unused filelists 2021-09-08 14:36:36 +00:00
Adrian Lancucki b169ad3ba1 [FastPitch/PyT] Update NGC checkpoint url 2021-09-07 07:27:31 -07:00
Tomasz Cheda 5d6d417ff5 [NCF/PyT] Adding BYOD capabilities 2021-09-03 06:20:12 -07:00
Lukasz Pierscieniewski 4fdd014ebf [Convnets] 21.09 Fixes 2021-09-02 07:35:20 -07:00