search

lucasjinreal / DCNv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)
BSD 3-Clause "New" or "Revised" License
637 stars 124 forks source link

run in mode CPU,results differente with GPU. fixed #74

Open dreamflywhere opened 8 months ago

dreamflywhere commented 8 months ago

run in mode CPU date of results not ok. you need modify dcnv2/src/cpu/dcn_v2_cpu.cpp

https://github.com/lucasjinreal/DCNv2_latest/blob/864c2850a780979f817c126b0e1a363a941ce07b/src/cpu/dcn_v2_cpu.cpp#L61

line 61: auto output = at::empty({batch, channels_out, height_out, width_out}, input.options()); modify:==> auto output = at::zeros({batch, channels_out, height_out, width_out}, input.options());