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
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
asulecki
cab1e33745
Fix DLLogger install
2021-11-02 15:12:02 +01: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
Michal Futrega
028534a5b9
[nnUNet/PyT] Add Jupyter notebook with BraTS21 solution
2021-10-20 10:03:05 -07:00
Szymon Migacz
11665f9cb2
[GNMT/PyT] requirements.txt: updated pytablewriter to 0.64.0
2021-10-13 09:40:51 -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
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
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
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
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
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