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

[Omnia Alwazzan]

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

Thanks

[Omnia Alwazzan]

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