Method __init__

Decoder Denoising Pretraining Model Args: lr: Learning rate optimizer: Name of optimizer (adam | adamw | sgd) betas: Adam beta parameters weight_decay: Optimizer weight decay momentum: SGD momentum parameter arch: Segme

(
        self,
        lr: float = 1e-4,
        optimizer: str = "adam",
        betas: tuple[float, float] = (0.9, 0.999),
        weight_decay: float = 0.0,
        momentum: float = 0.9,
        arch: str = "unet",
        encoder: str = "resnet18",
        in_channels: int = 3,
        mode: str = "decoder",
        noise_type: str = "scaled",
        noise_std: float = 0.22,
        loss_type: str = "l2",
        channel_last: bool = False,
    )