search

facebookresearch / ConvNeXt

Code release for ConvNeXt model
MIT License
5.78k stars 696 forks source link

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

Open omniaalwazzan opened 2 years ago

omniaalwazzan commented 2 years ago

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

Thanks

omniaalwazzan commented 2 years ago

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