Tan_pytorch_segmentation/pytorch_segmentation/PV_Backbone/ConvNeXtV2.py

from model.backbone.convnextv2 import convnextv2_atto
import torch
from torch import nn

if __name__ == "__main__":
    model = convnextv2_atto()
    input = torch.randn(1, 3, 224, 224)
    out = model(input)
    print(out.shape)