From 007e049922c32f87b812418a1b369aa38372a6fc Mon Sep 17 00:00:00 2001 From: Jakob Nybo Nissen Date: Thu, 27 Jun 2024 15:35:29 +0200 Subject: [PATCH] Tweak some default parameters Benchmarking has suggested than on the current version of Vamb, these parameters may be better. This needs additional benchmarking. --- vamb/__main__.py | 8 ++++---- vamb/encode.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/vamb/__main__.py b/vamb/__main__.py index 7591d608..613e8656 100755 --- a/vamb/__main__.py +++ b/vamb/__main__.py @@ -1783,8 +1783,8 @@ def add_vae_arguments(subparser): dest="beta", metavar="", type=float, - default=200.0, - help="beta, capacity to learn [200.0]", + default=500.0, + help="beta, capacity to learn [500.0]", ) vaeos.add_argument( "-d", @@ -1805,8 +1805,8 @@ def add_vae_arguments(subparser): dest="batchsize", metavar="", type=int, - default=256, - help="starting batch size [256]", + default=128, + help="starting batch size [128]", ) trainos.add_argument( "-q", diff --git a/vamb/encode.py b/vamb/encode.py index e9995e11..004dff28 100644 --- a/vamb/encode.py +++ b/vamb/encode.py @@ -184,7 +184,7 @@ def __init__( # If only 1 sample, we weigh alpha and nhiddens differently if alpha is None: - alpha = 0.15 if nsamples > 1 else 0.50 + alpha = 0.25 if nsamples > 1 else 0.50 if nhiddens is None: nhiddens = [512, 512] if nsamples > 1 else [256, 256]