From 97e5adf7cb56c53ca25a2192d452ddf372e02664 Mon Sep 17 00:00:00 2001
From: Pat Alt <55311242+pat-alt@users.noreply.github.com>
Date: Wed, 4 Oct 2023 12:59:47 +0200
Subject: [PATCH] cut down grid params for small datasets
---
experiments/setup_env.jl | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/experiments/setup_env.jl b/experiments/setup_env.jl
index a7b2108a..1f92ef11 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)],
)
--
GitLab