search

XiaLiPKU / EMANet

The code for Expectation-Maximization Attention Networks for Semantic Segmentation (ICCV'2019 Oral)
https://xialipku.github.io/publication/expectation-maximization-attention-networks-for-semantic-segmentation/
GNU General Public License v3.0
680 stars 130 forks source link

moving averaging operation #42

Open shuaizzZ opened 4 years ago

shuaizzZ commented 4 years ago

The moving averaging operation can also be writtern in EMAU class?

XiaLiPKU commented 4 years ago

With pytorch's new API (DDP included), you can also write it in EMAU class with the dist.reduce operation