From 8bd44d8fc7455fe778554929ee0402c0b8832e9c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?H=C3=A5vard=20Heitlo=20Holm?= Date: Wed, 6 Mar 2024 14:35:37 +0100 Subject: [PATCH] Better default parameters for KL model error --- src/gpuocean/utils/DoubleJetCase.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/src/gpuocean/utils/DoubleJetCase.py b/src/gpuocean/utils/DoubleJetCase.py index 9eaa659b..747768e5 100644 --- a/src/gpuocean/utils/DoubleJetCase.py +++ b/src/gpuocean/utils/DoubleJetCase.py @@ -211,13 +211,17 @@ def __init__(self, gpu_ctx, } elif model_error == 2: # KL model error - self.model_error_args = {"basis_x_start": 2, - "basis_x_end": 8, - "basis_y_start": 1, - "basis_y_end": 7, - "kl_decay": 1.25, - "kl_scaling": 0.025 - } + start_basis = 1 + num_bases = 7 + self.model_error_args = { + "basis_x_start": start_basis+1, + "basis_x_end": start_basis+1+num_bases, + "basis_y_start": start_basis, + "basis_y_end": start_basis+num_bases, + + "kl_decay": 0.9, + "kl_scaling": 0.001, + } else: assert(model_error in [0, 1, 2]), "Invalid value for model__error"