[Bug] SARNet half-precision error. #1441
Labels
framework: pytorch
Related to PyTorch backend
module: models
Related to doctr.models
topic: text recognition
Related to the task of text recognition
type: bug
Something isn't working
Milestone
Bug description
When use
sar_resnet31
I am getting this error:expected mat1 and mat2 to have the same dtype, but got: float != c10::Half
.But some other models (I tried
master
andcrnn_mobilenet_v3_large
) works well.Maybe somewhere in the sar_resnet31 architecture you missed to convert a tensor to half-precision. (imho)
Code snippet to reproduce the bug
The input image:
Error traceback
Environment
Actually what I got for the version built form the main branch.
The supposed env:
Deep Learning backend
is_tf_available: False
is_torch_available: True
The text was updated successfully, but these errors were encountered: