Skip to content

Config

helical.models.helix_mrna.HelixmRNAConfig

HelixmRNAConfig class to store the configuration of the Helix-mRNA model.

Parameters:

Name Type Description Default
batch_size int

The batch size

 10
device Literal['cpu', 'cuda']

The device to use. Either use "cuda" or "cpu".

 "cpu"
max_length int

The maximum length of the input sequence.

 12288
nproc int

Number of processes to use for data processing.

 1
Source code in helical/models/helix_mrna/helix_mrna_config.py 
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
class HelixmRNAConfig:
    """HelixmRNAConfig class to store the configuration of the Helix-mRNA model.

    Parameters
    ----------
    batch_size : int, optional, default=10
        The batch size
    device : Literal["cpu", "cuda"], optional, default="cpu"
        The device to use. Either use "cuda" or "cpu".
    max_length : int, optional, default=12288
        The maximum length of the input sequence.
    nproc: int, optional, default=1
        Number of processes to use for data processing.
    """

    def __init__(
        self,
        batch_size: int = 10,
        device: Literal["cpu", "cuda"] = "cpu",
        max_length: int = 12288,
        nproc: int = 1,
    ):

        model_name: Literal["helical-ai/Helix-mRNA"] = "helical-ai/Helix-mRNA"

        self.config = {
            "model_name": model_name,
            "input_size": max_length,
            "batch_size": batch_size,
            "device": device,
            "nproc": nproc,
        }