From d670dbaab55cdbbca40579a77371092d4b3ac9d2 Mon Sep 17 00:00:00 2001
From: Zohten <augustin.arnault@etu.imt-lille-douai.fr>
Date: Sun, 18 Jul 2021 17:58:06 +0200
Subject: [PATCH] replace len(list(x.size())) by x.dim()

---
 ranger21/ranger21.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ranger21/ranger21.py b/ranger21/ranger21.py
index a30b38c..bdf7b82 100644
--- a/ranger21/ranger21.py
+++ b/ranger21/ranger21.py
@@ -74,7 +74,7 @@ def get_chebs(num_epochs):
 
 def normalize_gradient(x, use_channels=False, epsilon=1e-8):
     """  use stdev to normalize gradients """
-    size = len(list(x.size()))
+    size = x.dim()
     # print(f"size = {size}")
 
     if (size > 1) and use_channels:
@@ -91,7 +91,7 @@ def normalize_gradient(x, use_channels=False, epsilon=1e-8):
 def centralize_gradient(x, gc_conv_only=False):
     """credit - https://github.com/Yonghongwei/Gradient-Centralization """
 
-    size = len(list(x.size()))
+    size = x.dim()
     # print(f"size = {size}")
 
     if gc_conv_only: