Commit graph

972 commits

Author SHA1 Message Date
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
Tomasz Grel 88e5298421 [VAE-CF-TF] Modify result logging 2021-08-31 16:53:43 +00:00
Andrzej Sulecki b2f30f365f [ConvNets/PyT] Minor fixes in README 2021-08-31 16:53:02 +00:00
Tomasz Grel 99ab780448 [DLRM/PyT] Fixed logging 2021-08-31 03:35:36 -07:00
Alexandre Milesi 72a90514f2 [SE3Transformer/PyT] Fix requirements on newer containers 2021-08-30 20:01:59 +00:00
Andrei Shumak 58df33d1a2 [SSD/PyT] Fixes 2021-08-30 12:47:32 -07:00
Ahmad Kiswani 248927e6fd [SSD-TF1] moved cython to requirements.txt 2021-08-27 14:19:49 +00:00
Alexandre Milesi (Engrg-Hardware 1) d74ff49e19 Add GNNs to main README 2021-08-27 14:18:46 +00:00
Sharath Turuvekere Sreenivas 78ec680207 Add GPus/Node in readme 2021-08-27 14:17:30 +00:00
Andrei Shumak d6cd6b869c [SSD/PyT] Improved logging 2021-08-27 07:16:50 -07:00
Subhankar Ghosh 01bbec9c4e [EfficientDet/PyT] Initial release 2021-08-25 08:23:18 -07:00
Adrian Lancucki ba8f04bfe2 [FastPitch/PyT] Add FastPitch 1.1 (autoaligned) 2021-08-23 22:52:23 +02:00
Alexandre Milesi (Engrg-Hardware 1) 85c54b5c36 [SE3Transformer/PyT] Public release and CI 2021-08-23 22:09:50 +02:00
kkudrynski b6e5ebdbc9 Typos and small fixes 2021-08-23 17:48:46 +02:00
Szymon Migacz ef98b2cef9
[TXL/PyT] update: (#989)
* changed API calls to torch.einsum
* added export OMP_NUM_THREADS=1 to all launcher scripts
* additional runtime checks to ensure that launch configuration is valid
2021-08-20 08:39:12 -07:00
Mikolaj Blaz 706ef498c9 [Jasper/PyT] Clean up inference flags 2021-08-13 19:25:20 +00:00
Mikolaj Blaz 568e246408 [Jasper/PyT] Add missing package requirements 2021-08-13 19:21:56 +00:00
Jan Baczek 233287038c [Transformer/PyT] Removing obsolete profiling code 2021-08-13 16:36:58 +00:00
gkarch 793027c9da removed triton stuff from model dockerfile and moved it to ./triton folder 2021-08-13 16:33:09 +00:00
Michal ee3f3db4c0 [nnUNet/PyT] update dlprof 2021-08-13 16:31:23 +00:00
Andrei Shumak 6a642837c4 Revert changes to FastPitch and BERT on PyTorch 2021-08-13 16:19:34 +02:00
Krzysztof Kudrynski a860701665 [BART/PyT] Initial release 2021-08-11 17:08:16 +00:00
Lukasz Pierscieniewski 69a26f5c76 [Multiple models] Added protocol specification to requirements.txt 2021-08-11 16:54:02 +00:00
Andrei Shumak 9c5766102f [PyT] BERT: new CI configurations 2021-08-06 15:18:29 +00:00
Andrei Shumak 9a6ccc9e23 Add MaskRCNN PyT new CI config support + tests
Update tools + regenerate all tests

Add one more configuration file

Update tools

Update configuration + update tools

check: remove `ntasks=1` for 8 gpu test

add --gpus-per-node=8

Upadte tools (task_per_gpu parameter added)

Update tools
2021-08-06 16:12:45 +02:00
Andrei Shumak 9ea704497c [PyT] FastPitch: new CI configurations 2021-08-06 14:08:33 +00:00
Andrzej Sulecki ab6267ad33 [internal/CI][PyT/ConvNets] Move to launcher 2021-07-26 21:18:30 +00:00
nv-kkudrynski d788e8d496
Merge pull request #970 from nv-kkudrynski/kkudrynski/readme_notice
Adding links to performance benchmark page
2021-07-21 14:46:11 +02:00
Krzysztof Kudrynski 49e23b4597 Adding links to performance benchmark page 2021-07-21 14:39:48 +02:00
Yuanzhe Dong 3d8d878489
[ConvNets/PyT] update triton repo url in readmes 2021-07-21 12:53:10 +02:00
Joaquin Anton 41e4a07a26
[Jasper/PyT] Update DALI Jasper pipeline to functional API
Signed-off-by: Joaquin Anton <janton@nvidia.com>
2021-07-21 12:47:05 +02:00
Michal ce7926fc80 [nnUNet/PyT] fix pyprof import 2021-07-19 20:02:36 +02:00
gkarch 5be8a3cae2 fix for github issue 965 2021-07-15 12:42:26 +00:00
Lukasz Pierscieniewski 33110132cc [Convnets/TF] Basic CPU model support 2021-07-15 08:22:43 +00:00
Andrzej Sulecki a1bbe6687e [PyT/ConvNets] Fixing bug in model2onnx 2021-07-12 10:20:43 +00:00