diff --git a/experiments/fmnist.jl b/experiments/fmnist.jl index 4c53693dcdb14d65b7e0f6141c55282cb760fc17..bb170d4d0a4df86b69d045aab44f86e6ee358cff 100644 --- a/experiments/fmnist.jl +++ b/experiments/fmnist.jl @@ -41,6 +41,7 @@ params = ( niter_eccco=10, Λ=[0.1, 0.25, 0.25], Λ_Δ=[0.1, 0.1, 2.5], + opt=Flux.Optimise.Descent(0.1) ) if !GRID_SEARCH diff --git a/experiments/mnist.jl b/experiments/mnist.jl index d9e73856ad6d08f1e7482196b1adcace74b827fd..f7fb962229536739dc320641478657863da6175e 100644 --- a/experiments/mnist.jl +++ b/experiments/mnist.jl @@ -41,6 +41,7 @@ params = ( niter_eccco=10, Λ=[0.1, 0.25, 0.25], Λ_Δ=[0.1, 0.1, 2.5], + opt=Flux.Optimise.Descent(0.1) ) if !GRID_SEARCH diff --git a/experiments/setup_env.jl b/experiments/setup_env.jl index 4852394c8522ce9bd8ff0440f0bc69e1b019a73d..e7a89092c189f901f29119507572db1207b2c6da 100644 --- a/experiments/setup_env.jl +++ b/experiments/setup_env.jl @@ -154,6 +154,11 @@ DEFAULT_GENERATOR_TUNING = ( [0.1, 0.5, 0.5], ], reg_strength=[0.0, 0.1, 0.5], + opt=[ + Flux.Optimise.Descent(0.1), + Flux.Optimise.Descent(0.05), + Flux.Optimise.Descent(0.01), + ], ) "Boolean flag to check if model tuning was specified."