search

lorenmt / reco

The implementation of "Bootstrapping Semantic Segmentation with Regional Contrast" [ICLR 2022].
https://shikun.io/projects/regional-contrast
Other
162 stars 25 forks source link

ASPPPooling issue #21

Closed fred-xue closed 2 years ago

fred-xue commented 2 years ago

Thanks for your code. I'm exploring your code, and ran into an issue with the line below. Seems not right. I'm on pytorch 1.10.

https://github.com/lorenmt/reco/blob/71c1b7324fd36049c6657b122ba3fef373877b72/network/deeplabv3/aspp.py#L37

lorenmt commented 2 years ago

Hello, I have tested all of the code with pytorch 1.7 before open-sourcing. So it might be from the version issue.

What's the exact problem you were facing?

fred-xue commented 2 years ago

thanks for the reply. It's my mistake in testing with a batch size of 1