import torch
from torch import nn
from transformers import PreTrainedModel, PretrainedConfig

class MODNetConfig(PretrainedConfig):
    model_type = "modnet"

    def __init__(self, backbone="mobilenetv2", **kwargs):
        super().__init__(**kwargs)
        self.backbone = backbone