Skip to content

Commit 85f03fe

Browse files
[Fix] paddle support stride, fix oom (#3413)
* paddle support stride, fix oom * refine
1 parent 55ba7d9 commit 85f03fe

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

Diff for: paddleseg/models/losses/semantic_connectivity_loss.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,10 @@ def compute_class_connectiveity(pred_conn, label_conn, pred_num_conn,
142142
pair_conn_num = 0
143143

144144
for j in range(1, pred_num_conn):
145-
pred_j_mask = pred_conn[:, :, j]
145+
if hasattr(paddle.Tensor, "contiguous"):
146+
pred_j_mask = pred_conn[:, :, j].contiguous()
147+
else:
148+
pred_j_mask = pred_conn[:, :, j]
146149
pred_j = pred_j_mask * pred
147150

148151
iou = compute_iou(pred_j, label_i, zero)

0 commit comments

Comments
 (0)