Module neuralops.config.model_config

Classes

class ModelConfig (name: str = 'distilbert-base-uncased',
num_labels: int = 2,
attention_type: str = 'full',
use_fast_tokenizer: bool = True,
trust_remote_code: bool = False,
cache_dir: str = './cache',
fix_layernorm_naming: bool = True,
variants: list[str] = <factory>)

Expand source code

@dataclass
class ModelConfig:
    name: str = "distilbert-base-uncased"
    num_labels: int = 2
    attention_type: str = "full"
    use_fast_tokenizer: bool = True
    trust_remote_code: bool = False
    cache_dir: str = "./cache"
    fix_layernorm_naming: bool = True
    variants: list[str] = field(default_factory=list)

ModelConfig(name: str = 'distilbert-base-uncased', num_labels: int = 2, attention_type: str = 'full', use_fast_tokenizer: bool = True, trust_remote_code: bool = False, cache_dir: str = './cache', fix_layernorm_naming: bool = True, variants: list[str] = )

Instance variables

var attention_type : str

The type of the None singleton.

var cache_dir : str

The type of the None singleton.

var fix_layernorm_naming : bool

The type of the None singleton.

var name : str

The type of the None singleton.

var num_labels : int

The type of the None singleton.

var trust_remote_code : bool

The type of the None singleton.

var use_fast_tokenizer : bool

The type of the None singleton.

var variants : list[str]

The type of the None singleton.