mirror of
https://github.com/RapidAI/RapidOCR.git
synced 2026-11-11 05:20:49 +08:00
* perf(torch): reduce rec logits on device before copying them to the CPU Signed-off-by: ExtReMLapin <3909752+ExtReMLapin@users.noreply.github.com> * refactor: keep __call__ returning raw outputs, add InferSession.infer_argmax for rec Signed-off-by: ExtReMLapin <3909752+ExtReMLapin@users.noreply.github.com> --------- Signed-off-by: ExtReMLapin <3909752+ExtReMLapin@users.noreply.github.com>