From aab030163c4e0a5fe2608b6b5a69471a7983853f Mon Sep 17 00:00:00 2001 From: Less Wright Date: Fri, 2 Jul 2021 17:05:25 -0700 Subject: [PATCH] extend agc to any dimension --- ranger21/ranger21.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/ranger21/ranger21.py b/ranger21/ranger21.py index dded932..8d1c1f5 100644 --- a/ranger21/ranger21.py +++ b/ranger21/ranger21.py @@ -379,9 +379,7 @@ class Ranger21(TO.Optimizer): elif xlen == 4: # conv kernels dim = (1, 2, 3) else: - raise ValueError( - f"unit_norm:: adaptive gclipping: unable to process len of {xlen} - currently must be <= 4" - ) + dim = tuple([x for x in range(1,xlen)]) # create 1,..., xlen-1 tuple, while avoiding last dim ... return x.norm(dim=dim, keepdim=keepdim, p=2.0)