Skip to content

Commit 2ae6e57

Browse files
committed
update
1 parent 3f80a40 commit 2ae6e57

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ppdet/modeling/necks/channel_mapper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def out_shape(self):
116116
def init_weights(self):
117117
"""Initialize the transformer weights."""
118118
for p in self.parameters():
119-
if p.rank() > 1:
119+
if p.dim() > 1:
120120
xavier_uniform_(p)
121121
if hasattr(p, 'bias') and p.bias is not None:
122122
constant_(p.bais)

0 commit comments

Comments
 (0)