Commit graph

25 commits

Author SHA1 Message Date
kkudrynski 3424cd03e2 [WideAndDeep/TF2] Release 2021-03-04 14:25:59 +01:00
kkudrynski 8e06f83dd9 [UNet medical/TF] Update to 21.02 2021-03-01 16:06:18 +01:00
kkudrynski 32e921cd99 [UNet medical/TF] Loss update 2021-02-02 15:52:32 +01:00
kkudrynski 8eca836a2e [UNet medical/TF2] minor fixes in paths 2021-02-02 14:38:49 +01:00
Shashank Verma 1f02ed1500 Add TF2 Segmentation README
Signed-off-by: Shashank Verma <shashank3959@gmail.com>
2020-12-02 00:50:05 -08:00
Shashank Verma 9f605f0e2f Add TF2 Language Modeling README
Signed-off-by: Shashank Verma <shashank3959@gmail.com>
2020-12-02 00:49:22 -08:00
kkudrynski 94a8f28554 [UNet medical/TF2] Fix 2020-11-23 17:18:27 +01:00
Jan Golda cca482832e Fixed mrcnn weights downloading script 2020-11-18 13:52:27 +01:00
kkudrynski 96e1700988 [ELECTRA/TF2] Pretraining and other updates 2020-10-26 12:19:47 +01:00
Michał Marcinkiewicz 751bca168d
Update README.md 2020-09-18 10:27:26 +02:00
kkudrynski 5d36b4fd2f Fixing hyperlinks 2020-09-08 16:10:32 +02:00
Sharath T S 446c878878
[ELECTRA/TF2] Update inference latency (#657)
* Update inference latency

* Fix inference perf numbers

* Fix latency computation
2020-08-19 20:43:44 -07:00
Pablo Ribalta Lorenzo fb40734b31
Remove autobench scripts (#647)
Signed-off-by: Pablo Ribalta <pribalta@nvidia.com>
2020-08-12 11:48:46 -07:00
Sharath T S 5a87078d25
[Electra/TF2] Fix container image and num_gpu arg (#628)
* Fix container image

* fix num_gpu arg
2020-07-30 14:48:34 -07:00
Pablo Ribalta f4625a2935 Update checkpoint MaskRCNN
Signed-off-by: Pablo Ribalta <pribalta@nvidia.com>
2020-07-23 15:00:13 +02:00
Amr Ragab 9ba22d1f29
[MaskRCNN/TF] Update extract_RN50_weights.py (#597)
* Update extract_RN50_weights.py

removing contrib - not compatible with TF2.0

* Update extract_RN50_weights.py

adding compat with tf1 and tf2
2020-07-22 17:01:27 -07:00
Przemek Strzelczyk b31d091cf3 [Electra/TF2] Adding new model 2020-07-20 20:39:35 +02:00
Przemek Strzelczyk b27abeba07 [UNet_medical/TF1&2] Updating for Ampere 2020-07-04 01:42:09 +02:00
Przemek Strzelczyk fa59dd505a [MaskRCNN/TF&PyT] Adding Ampere support 2020-06-27 11:52:08 +02:00
Michał Marcinkiewicz 57b8a6ac3a
Update losses.py
Replace softmax_cross_entropy_with_logits with binary_crossentropy
2020-04-26 10:24:08 +02:00
Michał Marcinkiewicz 1f04b9fe4d
Update Dockerfile 2020-04-13 19:32:09 +02:00
Pablo Ribalta Lorenzo c11a89aa76 Fix Unet tf2
Signed-off-by: Pablo Ribalta Lorenzo <pribalta@nvidia.com>
2020-04-02 07:20:58 +00:00
Przemek Strzelczyk f3a9f14bc7 [MaskRCNN/TF] Changelog fix 2020-03-05 09:50:43 +01:00
Przemek Strzelczyk 72ef5f45cb [MaskRCNN/TF2] Adding MaskRCNN for TF1 and TF2 2020-03-05 09:49:01 +01:00
Przemek Strzelczyk 0634210859 [UNet(med)/TF2] Adding TF2 version 2020-03-02 14:49:56 +01:00