From 2e74c9160c437eff16739db63197e6937b88b045 Mon Sep 17 00:00:00 2001 From: Connor Olding Date: Thu, 3 Aug 2017 03:39:25 +0000 Subject: [PATCH] tweak CubicGB defaults --- onn.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/onn.py b/onn.py index e429edc..f0053c4 100755 --- a/onn.py +++ b/onn.py @@ -470,7 +470,9 @@ class CubicGB(Layer): # paper: https://arxiv.org/abs/1707.04199 # note: in the paper, it's called pow3GB, which is ugly. - def __init__(self, alpha=0.001, beta=0.4): + def __init__(self, alpha=0.1, beta=0.0): + # note: the paper suggests defaults of 0.001 and 0.0, + # but these didn't seem to work as well in my limited testing. super().__init__() self.alpha = _f(alpha) self.beta = _f(beta)