Merge remote-tracking branch 'origin/norm_norm_norm' into bits_and_tpu

2 years ago · 3fce010ca8
parent 15cc9eae3e 0557c8257d
commit 3fce010ca8
1 changed files with 1 additions and 1 deletions
--- a/timm/optim/optim_factory.py
+++ b/timm/optim/optim_factory.py
@ -44,7 +44,7 @@ def param_groups_weight_decay(
        if not param.requires_grad:
            continue
-        if param.ndim or name.endswith(".bias") or name in no_weight_decay_list:
+        if param.ndim <= 1 or name.endswith(".bias") or name in no_weight_decay_list:
            no_decay.append(param)
        else:
            decay.append(param)