opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-09-24 23:33:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix amp train for re

Browse Source
This commit is contained in:
andyjpaddle
2022-08-08 09:02:11 +00:00
parent 1477b89aa4
commit 8e3aa3a09b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/program.py
+1 -1
View File
@@ -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