New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

BUG: segformer model fix embedding size #576

Open

valhassan opened this issue Aug 16, 2024 · 0 comments

Labels

bug

Collaborator

valhassan commented Aug 16, 2024

Describe the bug

I want to correct a fatal bug in the decoder architecture which implements the wrong embedding size in the decoder part of the segformer model.

geo-deep-learning/models/segformer.py

Lines 30 to 31 in fc633ee

    
           if encoder == "mit_b0" or "mit_b1": 
        
               embedding_dim = 256

valhassan added the bug label

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment