fix amp train for re

8e3aa3a0 · andyjpaddle · 1477b89a · 8e3aa3a0
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

tools/program.py tools/program.py +1 -1

未找到文件。
--- a/tools/program.py
+++ b/tools/program.py
@@ -171,7 +171,7 @@ def to_float32(preds):
            else:
                preds[k] = paddle.to_tensor(preds[k], dtype='float32')
    else:
-        preds[k] = paddle.to_tensor(preds[k], dtype='float32')
+        preds = paddle.to_tensor(preds, dtype='float32')
    return preds