File size: 322 Bytes
5189ac9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from transformers import PretrainedConfig
from typing import List
class DoduoConfig(PretrainedConfig):
model_type = "doduo"
def __init__(
self,
dino_corr_mask_ratio: float = 0.99,
**kwargs,
):
self.dino_corr_mask_ratio = dino_corr_mask_ratio
super().__init__(**kwargs) |