import torch.nn as nn class U2Net(nn.Module): def __init__(self): super(U2Net, self).__init__() pass