The model configurations for all the SegFormer B0 ~ B5

[rose-jinyang]

Hello
How are you?
Thanks for contributing to this project.
Is the model configuration in README MiT-B0 correctly?
That's because the total number of params for the model is 36M.
Could u provide all the model configurations for SegFormer B0 ~ B5?

[lucidrains]

@rose-jinyang hmm, i'm actually not too sure, i did go back and reread the paper and noticed they used depth-wise convs in the feedforward 5868d8e however, i don't think that's enough to account for the difference

[rose-jinyang]

I've updated the project to the new commit.
But the number of paprams for MiT-B0 is 7M.

[rose-jinyang]

Hi @lucidrains
Did u check the number of params of MiT-B0 model?

[rose-jinyang]

Hello @lucidrains
Did u check the number of params of MiT-B0 model?

[Sundragon1993]

Hi @rose-jinyang
Have you found the solution?