File size: 295 Bytes
4bdad8d | 1 2 3 4 5 6 7 8 9 10 11 | 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
|