wassname/Ranger21
1
0
Fork 0
mirror of https://github.com/wassname/Ranger21.git synced 2026-09-09 11:15:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

replace len(list(x.size())) by x.dim()

Browse Source
This commit is contained in:
Zohten
2021-07-18 17:58:06 +02:00
parent b575cc4cf3
commit d670dbaab5
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ranger21/ranger21.py
+2 -2
View File
@@ -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: