lucidrains / segformer-pytorch

Implementation of Segformer, Attention + MLP neural network for segmentation, in Pytorch
MIT License
342 stars 43 forks source link

batchNorm or layerNorm? #10

Open Napier7 opened 2 years ago

Napier7 commented 2 years ago

I see the author using batchNorm not layerNorm according to the mmsegmentation config file in the official depot. Am I misinterpreting this?