expected scalar type Float but found BFloat16

OpenGVLab / DCNv4

[CVPR 2024] Deformable Convolution v4

https://arxiv.org/pdf/2401.06197.pdf

MIT License

516 stars 27 forks source link

Closed jiarenyf closed 9 months ago

jiarenyf commented 10 months ago

DCNv4/functions/dcnv4_func.py", line 125, in backward ext.dcnv4_backward(*args) RuntimeError: expected scalar type Float but found BFloat16

JiaHeng-DLUT commented 3 months ago

+1. same problem.