refactor gradient filtering

2019-02-03 15:10:43 +01:00 · 2019-02-03 15:10:43 +01:00 · 54ea41711b
parent 0d28882ef0
commit 54ea41711b
1 changed files with 5 additions and 4 deletions
--- a/onn/optimizer.py
+++ b/onn/optimizer.py
@ -12,12 +12,13 @@ def filter_gradients(accum, grads, param):
    # param == 0 simply copies grads into accum.
    if param == 0:
        accum[:] = grads
-    if param < 0:
-        if param != -1:
-            accum *= -param
-        accum += grads
    elif param == 1:
        pass
+    elif param == -1:
+        accum += grads
+    elif param < 0:
+        accum *= -param
+        accum += grads
    else:
        accum += (1 - param) * (grads - accum)
    return accum