Update det_basic_loss.py

to_variable has been removed in paddle 2.0,using to_tensor instead
2021-02-05 15:49:01 +08:00 · 2021-02-05 15:49:01 +08:00 · 4a8fc9176d
parent a2fb623feb
commit 4a8fc9176d
1 changed files with 1 additions and 1 deletions
--- a/ppocr/losses/det_basic_loss.py
+++ b/ppocr/losses/det_basic_loss.py
@ -200,6 +200,6 @@ def ohem_batch(scores, gt_texts, training_masks, ohem_ratio):
                i, :, :], ohem_ratio))

    selected_masks = np.concatenate(selected_masks, 0)
-    selected_masks = paddle.to_variable(selected_masks)
+    selected_masks = paddle.to_tensor(selected_masks)

    return selected_masks