Skip to content
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.

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

convolution encoder better result then vit #14

Open
Alexandr1111111 opened this issue Jan 24, 2023 · 13 comments
Open

convolution encoder better result then vit #14

Alexandr1111111 opened this issue Jan 24, 2023 · 13 comments

Comments

@Alexandr1111111
Copy link

Alexandr1111111 commented Jan 24, 2023

Thank you for this work.
The generalizing ability of neural networks based on convolution layers is much greater.

class Model(torch.nn.Module):
    def __init__(self):
        super(Model, self).__init__()
        from efficientnet_pytorch import EfficientNet
        self.model = EfficientNet.from_pretrained('efficientnet-b4')
        self.model = EfficientNet.from_pretrained('efficientnet-b4')
        self.model._fc = torch.nn.Linear(1792, 1024)
        self.conv1D = torch.nn.Conv1d(1, 128, 3, padding='same')

    def forward(self, x):
        
        x = self.model(x)
        x = torch.unsqueeze(x, 1)
        x = self.conv1D(x)
        # return (batch, seq, dim)
        return x
@Alexandr1111111 Alexandr1111111 changed the title conv vs vit convolution encoder better result then vit Feb 12, 2023
@JackWhite-rwx
Copy link

thanks!could you give me a pretrained mdoel ?

@Alexandr1111111
Copy link
Author

Alexandr1111111 commented Mar 4, 2023 via email

@JackWhite-rwx
Copy link

thank you very much!

@Alexandr1111111
Copy link
Author

Alexandr1111111 commented Mar 8, 2023 via email

@JackWhite-rwx
Copy link

JackWhite-rwx commented Mar 9, 2023 via email

@Alexandr1111111
Copy link
Author

Alexandr1111111 commented Mar 9, 2023 via email

@JackWhite-rwx
Copy link

JackWhite-rwx commented Mar 9, 2023 via email

@Alexandr1111111
Copy link
Author

Alexandr1111111 commented Mar 9, 2023 via email

@JackWhite-rwx
Copy link

JackWhite-rwx commented Mar 9, 2023 via email

@Alexandr1111111
Copy link
Author

Alexandr1111111 commented Mar 9, 2023 via email

@Alexandr1111111
Copy link
Author

Alexandr1111111 commented Mar 13, 2023 via email

@JackWhite-rwx
Copy link

JackWhite-rwx commented Mar 14, 2023 via email

@Alexandr1111111
Copy link
Author

Alexandr1111111 commented Mar 14, 2023 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants