default ema decay to 0.5

lucidrains · Feb 27, 2021 · 3b2e1a0 · 3b2e1a0
1 parent 51ac409
commit 3b2e1a0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/big_sleep/cli.py b/big_sleep/cli.py
@@ -24,7 +24,7 @@ def train(
     class_temperature = 2.,
     save_best = False,
     experimental_resample = False,
-    ema_decay = 0.99
+    ema_decay = 0.5
 ):
     print(f'Starting up... v{__version__}')
 

diff --git a/big_sleep/version.py b/big_sleep/version.py
@@ -1 +1 @@
-__version__ = '0.6.1'
+__version__ = '0.6.2'