Skip to content

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

@dreamflywhere

Description

@dreamflywhere

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

auto output = at::empty({batch, channels_out, height_out, width_out}, input.options());

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());

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions