diff --git a/experiments/setup_env.jl b/experiments/setup_env.jl index a7b2108a8c89bd313c5bae222778cb9c76c6bd55..1f92ef116aa53eef8ffe439a7304e34a49ba5c0f 100644 --- a/experiments/setup_env.jl +++ b/experiments/setup_env.jl @@ -182,12 +182,13 @@ const GRID_SEARCH = "grid_search" ∈ ARGS "Generator tuning parameters." DEFAULT_GENERATOR_TUNING = ( - Λ = [[0.1, 0.1, 0.05], [0.1, 0.1, 0.1], [0.1, 0.1, 0.5], [0.1, 0.1, 1.0]], - reg_strength = [0.0, 0.1, 0.25, 0.5, 1.0], + Λ=[[0.1, 0.1, 0.1], [0.1, 0.1, 0.5],], + reg_strength = [0.0, 0.1, 1.0], opt = [ Descent(0.01), Descent(0.05), - Descent(0.1), + Optimiser(ClipValue(0.01), Descent(0.01)), + Optimiser(ClipValue(0.05), Descent(0.05)), ], decay = [(0.0, 1), (0.01, 1), (0.1, 1)], )