Fix warning from tfruns::training_run() being unable to log optimiz…

…er learning rate.
rstudio · Jul 3, 2024 · a965e04 · a965e04
1 parent 4cd188b
commit a965e04
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/NEWS.md b/NEWS.md
@@ -5,6 +5,8 @@
 
 - `keras_shape` objects (as returned by `keras3::shape()`) gain `==` and `!=` methods.
 
+- Fixed warning from `tfruns::training_run()` being unable to log optimizer learning rate.
+
 - Added compatibility with Keras v3.4.1 (no R user facing changes).
 
 User facing changes with upstream Keras v3.4.0:

diff --git a/R/zzz-metrics-callback.R b/R/zzz-metrics-callback.R
@@ -119,7 +119,7 @@ callback_view_metrics <- Callback(
         optimizer <- model$optimizer
         if (!is.null(optimizer)) {
           model_info$optimizer <- py_str(optimizer)
-          model_info$learning_rate <- as.double(optimizer$lr)
+          model_info$learning_rate <- as.double(optimizer$learning_rate)
         }
         tfruns::write_run_metadata("properties", model_info)
       }, error = function(e) {