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

Merge pull request #7139 from andyjpaddle/fix_amp_re

Browse Source 
[TIPC] Fix amp train for re
This commit is contained in:
zhoujun
2022-08-08 19:31:54 +08:00
committed by GitHub
parent 44852aae0c 8e3aa3a09b
commit 6445362fdc
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/program.py
+5 -3
View File
@@ -154,13 +154,14 @@ def check_xpu(use_xpu):
except Exception as e:
pass
def to_float32(preds):
if isinstance(preds, dict):
for k in preds:
if isinstance(preds[k], dict) or isinstance(preds[k], list):
preds[k] = to_float32(preds[k])
else:
preds[k] = preds[k].astype(paddle.float32)
preds[k] = paddle.to_tensor(preds[k], dtype='float32')
elif isinstance(preds, list):
for k in range(len(preds)):
if isinstance(preds[k], dict):
@@ -168,11 +169,12 @@ def to_float32(preds):
elif isinstance(preds[k], list):
preds[k] = to_float32(preds[k])
else:
preds[k] = preds[k].astype(paddle.float32)
preds[k] = paddle.to_tensor(preds[k], dtype='float32')
else:
preds = preds.astype(paddle.float32)
preds = paddle.to_tensor(preds, dtype='float32')
return preds
def train(config,
train_dataloader,
valid_dataloader,