search

suriyachaudary / Self-supervision-for-segmenting-overhead-imagery

[BMVC 2018] Self-supervised Feature Learning for Semantic Segmentation of Overhead Imagery.
MIT License
40 stars 7 forks source link

I think there is some error in your codes. #7

Closed tianjiu233 closed 4 years ago

tianjiu233 commented 4 years ago

In your codes, the fcn traincoah there is a line that outputs = net(inputs*masks).detach()

If the detach() is added, the backward() will not work. And the only loss will only be KLD. Is is right or there is something else I do not know?

Hope for your reply!