fix lr argument

xkykai · Nov 7, 2024 · 1758ca4 · 1758ca4
1 parent 3b91259
commit 1758ca4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...Qb_Ri_nof_BBLRifirst_NN5_localbaseclosure_convectivetanh_2Pr_shearlinear_wTwS_nouvloss.jl b/...Qb_Ri_nof_BBLRifirst_NN5_localbaseclosure_convectivetanh_2Pr_shearlinear_wTwS_nouvloss.jl
@@ -962,7 +962,7 @@ end
 # end_epochs = cumsum(maxiters)
 # training_timeframes = [timeframes[1][1:5], timeframes[1][1:10], timeframes[1][1:15], timeframes[1][1:20], timeframes[1][1:25], timeframes[1][1:27]]
 
-optimizers = [Optimisers.Adam(3e-4), Optimisers.Adam(3e-4), Optimisers.Adam(3e-4)]
+optimizers = [Optimisers.Adam(learning_rate), Optimisers.Adam(learning_rate), Optimisers.Adam(learning_rate)]
 maxiters = [5000, 5000, 5000]
 end_epochs = cumsum(maxiters)
 training_timeframes = [timeframes[1][1:10], timeframes[1][1:20], timeframes[1][1:27]]