Skip to content

GitLab

Switch branch/tag
  1. 06 Apr, 2021 3 commits
    • Reed Wanderman-Milne's avatar
      Use tf.compat.v1 version of enable_mixed_precision_graph_rewrite. · 920defcc
      Reed Wanderman-Milne authored 
      The function `tf.train.experimental.enable_mixed_precision_graph_rewrite` will be removed from the TF2 namespace soon, at which point it will only be accessible under tf.compat.v1.

PiperOrigin-RevId: 367046393
      920defcc
    • A. Unique TensorFlower's avatar
      Internal change · 17814043
      A. Unique TensorFlower authored 
      PiperOrigin-RevId: 366939051
      17814043
    • Jeremiah Liu's avatar
      Disable temperature scaling during training. · ff3ed4cc
      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
      ff3ed4cc
  3. 05 Apr, 2021 10 commits
  5. 03 Apr, 2021 1 commit
  7. 02 Apr, 2021 5 commits
  9. 01 Apr, 2021 6 commits
  11. 31 Mar, 2021 2 commits
  13. 30 Mar, 2021 5 commits
  15. 29 Mar, 2021 6 commits
  17. 26 Mar, 2021 2 commits