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
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
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
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
Elton Chen-Yu Ho
a5f26eeef7
Set GPU index based on local_rank instead of rank
2021-08-22 17:45:54 +08: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