import torch
import torch.nn as nn


class BaseModel(nn.Module):
    """ Base Model for implementation different models.
    """

    def __init__(self, activation: nn.Module, in_ch: int=1):
        super().__init__()

import torch
import torch.nn as nn


class BaseModel(nn.Module):
    """ Base Model for implementation different models.
    """

    def __init__(self, activation: nn.Module, in_ch: int=1):
        super().__init__()

Base Model for implementation different models.