bugfix save_mode

maximilian-gelbrecht · Aug 6, 2024 · 55789ac · 55789ac
1 parent cebb754
commit 55789ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/training.jl b/src/training.jl
@@ -123,7 +123,7 @@ function train!(model, ps, st, loss_func, train_data, opt_state, η_schedule; τ
                 end
             elseif save_mode==:additional_metric 
                 if gf[:forecast_delta][end] < lowest_additional_metric
-                    lowest_additional_metric = gf[end]
+                    lowest_additional_metric = gf[:forecast_delta][end]
                     best_ps = deepcopy(ps)
                     results[:loss_min] .= lowest_additional_metric
                     results[:i_epoch_min] .= i_epoch