This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

head layer can't be replaced in convnex like other(classifier layer) pytorch model? #129

Open

omniaalwazzan opened this issue Oct 17, 2022 · 1 comment

omniaalwazzan commented Oct 17, 2022

How to change the head layer like other pytorch models for example using nn.Sequential(...)

Thanks

Author

omniaalwazzan commented Oct 18, 2022

I have fixed the problem by using convnext that is under torchvsion.models instead of timm.

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