fix learning rate

xkykai · Nov 14, 2024 · 646130a · 646130a
1 parent 9aae214
commit 646130a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...calbaseclosure_convectivetanh_shearlinear_2Pr_freeconvection_windvalidation_multichunk.jl b/...calbaseclosure_convectivetanh_shearlinear_2Pr_freeconvection_windvalidation_multichunk.jl
@@ -1179,7 +1179,7 @@ function train_NDE_multipleics(ps, params, ps_baseclosure, sts, NNs, truths, x
     return ps_min, ps_min_validation, (; total=losses, total_validation=losses_validation), opt_statemin, opt_statemin_validation, iter_min, iter_min_validation
 end
 
-optimizers = [Optimisers.Adam(3e-4), Optimisers.Adam(learning_rate), Optimisers.Adam(learning_rate), Optimisers.Adam(learning_rate)]
+optimizers = [Optimisers.Adam(3e-4), Optimisers.Adam(3e-5), Optimisers.Adam(3e-5), Optimisers.Adam(3e-5)]
 maxiters = [2000, 2000, 2000, 2000]
 end_epochs = cumsum(maxiters)