search

chengdazhi / Deformable-Convolution-V2-PyTorch

Deformable ConvNets V2 (DCNv2) in PyTorch
MIT License
1.43k stars 229 forks source link

Not supported for apex #45

Open ming71 opened 4 years ago

ming71 commented 4 years ago

I've tried to train model with DCNv2 via apex mix-precision method, but occurred a dtype bug:

"deform_conv_forward_cuda" not implemented for 'Half'

Does it mean that I cannot train model with half precision? thx

cmsfw-github commented 4 years ago

I encounter the same problem

KosukeHao commented 4 years ago

I have the same problem