import torch
from src.backbone.senet import get_seresnet


def test_get_seresnet():
    x = torch.rand([4, 3, 224, 224])
    model = get_seresnet('seresnet50', nclass=100)
    assert list(model(x).shape) == [4, 100]

47915e23bb0b704988c71ddddaad0e6b282c712a
<mask token>

<mask token>


def test_get_seresnet():
    x = torch.rand([4, 3, 224, 224])
    model = get_seresnet('seresnet50', nclass=100)
    assert list(model(x).shape) == [4, 100]

import torch
from src.backbone.senet import get_seresnet


def test_get_seresnet():
    x = torch.rand([4, 3, 224, 224])
    model = get_seresnet('seresnet50', nclass=100)
    assert list(model(x).shape) == [4, 100]
