from transformers.configuration_utils import PretrainedConfig


class FCN4FlareConfig(PretrainedConfig):
    """
    Configuration class for FCN4Flare model.
    """
    model_type = "fcn4flare"

    def __init__(
        self,
        input_dim=3,
        hidden_dim=64,
        output_dim=1,
        depth=4,
        dilation=[1, 2, 4, 8],
        maskdice_threshold=0.5,
        dropout_rate=0.1,
        kernel_size=3,
        **kwargs
    ):
        """Initialize FCN4FlareConfig."""
        super().__init__(**kwargs)
        
        self.input_dim = input_dim
        self.hidden_dim = hidden_dim
        self.output_dim = output_dim
        self.depth = depth
        self.dilation = dilation
        self.maskdice_threshold = maskdice_threshold
        self.dropout_rate = dropout_rate
        self.kernel_size = kernel_size