- 05 Apr, 2021 11 commits
-
-
Reed Wanderman-Milne authored
This shouldn't break any official models, since I changed all LossScaleOptimizer isinstance checks to use the nonexperimental version (the experimental LSO subclasses the nonexperimental LSO, so changing isinstance checks in this way is always safe). PiperOrigin-RevId: 366891847
-
Jeremiah Liu authored
This change allows `GaussianProcessClassificationHead` to output only the predictive logits during training and evaluation (instead of outputting a tuple `(logits, covmat)`). The goal is to make the layer more compatible with `SentencePredictionTask` and Keras' `model.fit()` API. PiperOrigin-RevId: 366891298
-
Fan Yang authored
PiperOrigin-RevId: 366889393
-
Allen Wang authored
PiperOrigin-RevId: 366883662
-
Hongkun Yu authored
PiperOrigin-RevId: 366883391
-
Hongkun Yu authored
PiperOrigin-RevId: 366883351
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 366879385
-
A. Unique TensorFlower authored
Rewrite the GIOU loss op in matrix form to avoid the map_fn. map_fn could be prohibitively costly when dealing with large number of bounding boxes. PiperOrigin-RevId: 366854232
-
Vighnesh Birodkar authored
PiperOrigin-RevId: 366850188
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 366819679
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 366817283
-
- 04 Apr, 2021 1 commit
-
-
Pablo Ribalta Lorenzo authored
Signed-off-by:Pablo Ribalta Lorenzo <pribalta@nvidia.com>
-
- 03 Apr, 2021 1 commit
-
-
Yeqing Li authored
PiperOrigin-RevId: 366540340
-
- 02 Apr, 2021 9 commits
-
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 366496850
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 366395416
-
Vighnesh Birodkar authored
PiperOrigin-RevId: 366395416
-
Abdullah Rashwan authored
PiperOrigin-RevId: 366391425
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 366382220
-
Vighnesh Birodkar authored
PiperOrigin-RevId: 366388829
-
Vighnesh Birodkar authored
PiperOrigin-RevId: 366385685
-
Vighnesh Birodkar authored
PiperOrigin-RevId: 366382220
-
Frederick Liu authored
PiperOrigin-RevId: 366368802
-
- 01 Apr, 2021 6 commits
-
-
Chen Chen authored
PiperOrigin-RevId: 366354480
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 366330392
-
Chen Chen authored
PiperOrigin-RevId: 366313554
-
Jeremiah Liu authored
PiperOrigin-RevId: 366178203
-
Jeremiah Liu authored
PiperOrigin-RevId: 366175727
-
Jeremiah Liu authored
PiperOrigin-RevId: 366170373
-
- 31 Mar, 2021 3 commits
-
-
Ronny Votel authored
PiperOrigin-RevId: 366134194
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 366120838
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 366052172
-
- 30 Mar, 2021 8 commits
-
-
Abdullah Rashwan authored
PiperOrigin-RevId: 365926330
-
Abdullah Rashwan authored
PiperOrigin-RevId: 365914010
-
Austin Myers authored
PiperOrigin-RevId: 365913722
-
Austin Myers authored
PiperOrigin-RevId: 365899795
-
A. Unique TensorFlower authored
This CL register a pre-defined GIOU loss to the losses.proto and losses_builder so that the GIOU loss can be correctly called & built when GIOU loss is specified in the loss config file. PiperOrigin-RevId: 365891022
-
Fan Yang authored
PiperOrigin-RevId: 365858684
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 365828860
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 365713370
-
- 29 Mar, 2021 1 commit
-
-
Jonathan Huang authored
PiperOrigin-RevId: 365683210
-