from torch.nn import Module


class CNNModel(Module):

    def __init__(self) -> None:
        super(CNNModel, self).__init__()

from torch.nn import Module


class CNNModel(Module):

    def __init__(self) -> None:
        super(CNNModel, self).__init__()
