[Cherry-pick] #1579 (#1580)

PaddlePaddle · Dec 1, 2021 · 7e37f98 · 7e37f98
1 parent 5e6cab5
commit 7e37f98
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddleseg/models/backbones/mix_transformer.py b/paddleseg/models/backbones/mix_transformer.py
@@ -308,7 +308,7 @@ def __init__(self,
 
         # transformer encoder
         dpr = [
-            x.item() for x in paddle.linspace(0, drop_path_rate, sum(depths))
+            x.numpy() for x in paddle.linspace(0, drop_path_rate, sum(depths))
         ]  # stochastic depth decay rule
         cur = 0
         self.block1 = nn.LayerList([