semantic-pyramid-pytorch icon indicating copy to clipboard operation
semantic-pyramid-pytorch copied to clipboard

mask in reconstruction loss

Open jychoi118 opened this issue 4 years ago • 1 comments

I think your reconstruction loss is missing mask, which should be applied after subtracting two features as in (3) of the paper. image

https://github.com/rosinality/semantic-pyramid-pytorch/blob/6d4796cbb2abdbe9afaafb8dae75b60a3b619d2d/train.py#L146

jychoi118 avatar Jun 08 '20 04:06 jychoi118

Oh, thank you! Fixed it in d3777bf.

rosinality avatar Jun 19 '20 12:06 rosinality