File size: 364 Bytes
695da21 769be92 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | from transformers.configuration_utils import PretrainedConfig
class BASNetConfig(PretrainedConfig):
model_type = "basnet"
def __init__(
self,
n_channels: int = 3,
kernel_size: int = 3,
**kwargs,
) -> None:
super().__init__(**kwargs)
self.n_channels = n_channels
self.kernel_size = kernel_size
|