File size: 447 Bytes
cdd61a2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from transformers.configuration_utils import PretrainedConfig
class BASNetConfig(PretrainedConfig):
model_type = "basnet"
def __init__(
self,
resnet_model: str = "microsoft/resnet-34",
n_channels: int = 3,
kernel_size: int = 3,
**kwargs,
) -> None:
super().__init__(**kwargs)
self.resnet_model = resnet_model
self.n_channels = n_channels
self.kernel_size = 3
|