deepfold.config.EvoformerStackConfig¶

class deepfold.config.EvoformerStackConfig(c_m: 'int' = 256, c_z: 'int' = 128, c_hidden_msa_att: 'int' = 32, c_hidden_opm: 'int' = 32, c_hidden_tri_mul: 'int' = 128, c_hidden_tri_att: 'int' = 32, c_s: 'int' = 384, num_heads_msa: 'int' = 8, num_heads_tri: 'int' = 4, num_blocks: 'int' = 48, transition_n: 'int' = 4, msa_dropout: 'float' = 0.15, pair_dropout: 'float' = 0.25, inf: 'float' = 1000000000.0, eps_opm: 'float' = 0.001, chunk_size_msa_att: 'Optional[int]' = None, chunk_size_opm: 'Optional[int]' = None, chunk_size_tri_att: 'Optional[int]' = None, block_size_tri_mul: 'Optional[int]' = None, outer_product_mean_first: 'bool' = False)[source]¶

__init__(c_m: int = 256, c_z: int = 128, c_hidden_msa_att: int = 32, c_hidden_opm: int = 32, c_hidden_tri_mul: int = 128, c_hidden_tri_att: int = 32, c_s: int = 384, num_heads_msa: int = 8, num_heads_tri: int = 4, num_blocks: int = 48, transition_n: int = 4, msa_dropout: float = 0.15, pair_dropout: float = 0.25, inf: float = 1000000000.0, eps_opm: float = 0.001, chunk_size_msa_att: int | None = None, chunk_size_opm: int | None = None, chunk_size_tri_att: int | None = None, block_size_tri_mul: int | None = None, outer_product_mean_first: bool = False) → None¶

Methods

__init__([c_m, c_z, c_hidden_msa_att, ...])

Attributes

`block_size_tri_mul`
`c_hidden_msa_att`
`c_hidden_opm`
`c_hidden_tri_att`
`c_hidden_tri_mul`
`c_m`
`c_s`
`c_z`
`chunk_size_msa_att`
`chunk_size_opm`
`chunk_size_tri_att`
`eps_opm`
`inf`
`msa_dropout`
`num_blocks`
`num_heads_msa`
`num_heads_tri`
`outer_product_mean_first`
`pair_dropout`
`transition_n`