- 09 Apr, 2021 1 commit
-
-
Reed Wanderman-Milne authored
All models which support loss scaling support dynamic loss scaling, so the argument has no purpose. It used to be that some models scaled the loss manually instead of using a LossScaleOptimizer, and so did not support dynamic loss scaling. PiperOrigin-RevId: 367719521
-
- 08 Apr, 2021 1 commit
-
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 367463455
-
- 07 Apr, 2021 1 commit
-
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 367244904
-
- 06 Apr, 2021 2 commits
-
-
Hongkun Yu authored
PiperOrigin-RevId: 367101911
-
Jeremiah Liu authored
For the `GaussianProcessClassificationHead`, the temperature scaling needs to be disabled during training to avoid unexpected modification to the learning rate, which harms model quality. (Unfortunately, this seems to require adding `training` to the `call` method). Also set the default of `gp_cov_ridge_penalty` in `RandomFeatureGaussianProcess` to 1 to be consistent with that in the `GaussianProcessClassificationHead`. PiperOrigin-RevId: 366917075
-
- 05 Apr, 2021 2 commits
-
-
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
-
Hongkun Yu authored
PiperOrigin-RevId: 366883351
-
- 01 Apr, 2021 5 commits
-
-
Chen Chen authored
PiperOrigin-RevId: 366354480
-
Chen Chen authored
PiperOrigin-RevId: 366313554
-
Jeremiah Liu authored
PiperOrigin-RevId: 366178203
-
Jeremiah Liu authored
PiperOrigin-RevId: 366175727
-
Jeremiah Liu authored
PiperOrigin-RevId: 366170373
-
- 29 Mar, 2021 1 commit
-
-
Zachary Nado authored
adding `beta_1` parameter to `create_optimizer` in `optimization.py` to allow for a tunable `beta_1`. Defaults to the original `0.9` to ensure existing compatibility. PiperOrigin-RevId: 365634937
-
- 26 Mar, 2021 3 commits
-
-
Frederick Liu authored
PiperOrigin-RevId: 365254703
-
Frederick Liu authored
PiperOrigin-RevId: 365242134
-
Frederick Liu authored
PiperOrigin-RevId: 365188568
-
- 25 Mar, 2021 1 commit
-
-
Jeremiah Liu authored
PiperOrigin-RevId: 364952108
-
- 24 Mar, 2021 1 commit
-
-
Jeremiah Liu authored
PiperOrigin-RevId: 364722225
-
- 23 Mar, 2021 1 commit
-
-
Hongkun Yu authored
PiperOrigin-RevId: 364634581
-
- 22 Mar, 2021 4 commits
-
-
Poorva Potdar authored
PiperOrigin-RevId: 364378436
-
Jeremiah Liu authored
PiperOrigin-RevId: 364328905
-
Jeremiah Liu authored
PiperOrigin-RevId: 364226289
-
Le Hou authored
PiperOrigin-RevId: 364219111
-
- 20 Mar, 2021 2 commits
-
-
Jeremiah Liu authored
PiperOrigin-RevId: 364109398
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 364096705
-
- 18 Mar, 2021 1 commit
-
-
Hongkun Yu authored
PiperOrigin-RevId: 363593694
-
- 17 Mar, 2021 1 commit
-
-
A. Unique TensorFlower authored
both are tracked in the same bug. PiperOrigin-RevId: 363431367
-
- 16 Mar, 2021 3 commits
-
-
Frederick Liu authored
PiperOrigin-RevId: 363233691
-
Frederick Liu authored
PiperOrigin-RevId: 363121226
-
Hongkun Yu authored
PiperOrigin-RevId: 363069319
-
- 14 Mar, 2021 1 commit
-
-
- 12 Mar, 2021 3 commits
-
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 362613992
-
Chen Chen authored
PiperOrigin-RevId: 362610475
-
Chen Chen authored
PiperOrigin-RevId: 362461011
-
- 10 Mar, 2021 1 commit
-
-
Frederick Liu authored
PiperOrigin-RevId: 361957289
-
- 09 Mar, 2021 1 commit
-
-
Chen Chen authored
PiperOrigin-RevId: 361916017
-
- 08 Mar, 2021 1 commit
-
-
Chen Chen authored
PiperOrigin-RevId: 361482425
-
- 05 Mar, 2021 1 commit
-
-
Hongkun Yu authored
PiperOrigin-RevId: 361181257
-
- 04 Mar, 2021 2 commits
-
-
Poorva Potdar authored
PiperOrigin-RevId: 360985819
-
A. Unique TensorFlower authored
PiperOrigin-RevId: 360932205
-