Skip to content

Commit a7e05bc

Browse files
authored
changed x.numpy to x.item in mix_transformer (#3891)
1 parent fe5d70e commit a7e05bc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddleseg/models/backbones/mix_transformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -305,7 +305,7 @@ def __init__(self,
305305

306306
# transformer encoder
307307
dpr = [
308-
x.numpy() for x in paddle.linspace(0, drop_path_rate, sum(depths))
308+
x.item() for x in paddle.linspace(0, drop_path_rate, sum(depths))
309309
] # stochastic depth decay rule
310310
cur = 0
311311
self.block1 = nn.LayerList([

0 commit comments

Comments
 (0)