Params

Param dataclass

A dataclass representing a parameter in a checkpoint.

Attributes:

Name	Type	Description
name	str	The name of the parameter in the checkpoint.
partition_dim	int	The dimension index that gets sharded. None for no sharding.
hidden_dim	int	The hidden dimension index. None for no hidden dimension.

Source code in bionemo/evo2/utils/checkpoint/params.py

@dataclass
class Param:
    """A dataclass representing a parameter in a checkpoint.

    Attributes:
        name (str): The name of the parameter in the checkpoint.
        partition_dim (int): The dimension index that gets sharded. `None` for no sharding.
        hidden_dim (int): The hidden dimension index. `None` for no hidden dimension.
    """

    name: str  # Name of the parameter in the checkpoint.
    partition_dim: int  # The dimension index that gets sharded. `None` for no sharding.
    hidden_dim: int  # The hidden dimension index. `None` for no hidden dimension.