File size: 835 Bytes
0d2aee9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
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
|