Skip to content

GitLab

Switch branch/tag
  1. 28 May, 2021 1 commit
  3. 13 Apr, 2021 2 commits
    • Reed Wanderman-Milne's avatar
      Use nonexperimental mixed precision API. · 9a4d14a9
      Reed Wanderman-Milne authored 
      This replaces symbols in tf.keras.mixed_precision.experimental with the corresponding nonexperimental symbols. In some cases, passing a Policy is replaced with passing a policy name for conciseness.

Additionally, for the Shakespeare model, the loss_scale flag is removed, since supporting it with the nonexperimental API is slightly more verbose and it is recommended users use the default loss scale.

PiperOrigin-RevId: 368123944
      9a4d14a9
    • Reed Wanderman-Milne's avatar
      Use nonexperimental mixed precision API. · 4334a892
      Reed Wanderman-Milne authored 
      This replaces symbols in tf.keras.mixed_precision.experimental with the corresponding nonexperimental symbols. In some cases, passing a Policy is replaced with passing a policy name for conciseness.

Additionally, for the Shakespeare model, the loss_scale flag is removed, since supporting it with the nonexperimental API is slightly more verbose and it is recommended users use the default loss scale.

PiperOrigin-RevId: 368123944
      4334a892
  5. 29 Mar, 2021 2 commits
  7. 22 Mar, 2021 2 commits
  9. 05 Mar, 2021 2 commits
  11. 21 Dec, 2020 1 commit
  13. 02 Dec, 2020 2 commits
  15. 30 Nov, 2020 4 commits
  17. 27 Nov, 2020 2 commits
  19. 26 Nov, 2020 4 commits
  21. 24 Nov, 2020 2 commits
  23. 03 Nov, 2020 2 commits
  25. 13 Sep, 2020 2 commits
  27. 10 Sep, 2020 2 commits
  29. 12 Aug, 2020 2 commits
  31. 04 Aug, 2020 2 commits
  33. 03 Aug, 2020 2 commits
  35. 31 Jul, 2020 3 commits
  37. 30 Jul, 2020 1 commit
    • Tomer Kaftan's avatar
      Pre-emptively disable the KerasTensors refactoring for the detection models in... · 071bb243
      Tomer Kaftan authored 
      Pre-emptively disable the KerasTensors refactoring for the detection models in tensorflow models/official/vision/detection, because they rely on several unsupported things that will stop working entirely when the refactoring goes live.

Specifically:
* The custom layers implement `__call__` instead of `call` and rely on manually enter the keras backend graph
* The vision models try to use `tf.while_loop` as Keras op layers during functional API construction, which is unsupported.

Updating the models to avoid this would subtly change the variable names and break the pre-existing tf1-style name-based checkpoints, so for now we will just disable the KerasTensors refactoring for these models.

PiperOrigin-RevId: 323937426
      071bb243