Commit graph

14 commits

Author SHA1 Message Date
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