gm.ckpts.CheckpointPath

class gemma.gm.ckpts.CheckpointPath(value)

基类: enum.StrEnum

Gemma 检查点的硬编码路径。

格式: {VERSION}_{SIZE}_{VARIANT}。

变体

• PT: 预训练

• IT: 指令调优 (很可能是您想要使用的模型)

例如，GEMMA2_27B_IT 是 Gemma V2，270 亿参数，指令调优。

GEMMA2_2B_PT = 'gs://gemma-data/checkpoints/gemma2-2b-pt'

GEMMA2_9B_PT = 'gs://gemma-data/checkpoints/gemma2-9b-pt'

GEMMA2_27B_PT = 'gs://gemma-data/checkpoints/gemma2-27b-pt'

GEMMA2_2B_IT = 'gs://gemma-data/checkpoints/gemma2-2b-it'

GEMMA2_9B_IT = 'gs://gemma-data/checkpoints/gemma2-9b-it'

GEMMA2_27B_IT = 'gs://gemma-data/checkpoints/gemma2-27b-it'

GEMMA3_1B_PT = 'gs://gemma-data/checkpoints/gemma3-1b-pt'

GEMMA3_4B_PT = 'gs://gemma-data/checkpoints/gemma3-4b-pt'

GEMMA3_12B_PT = 'gs://gemma-data/checkpoints/gemma3-12b-pt'

GEMMA3_27B_PT = 'gs://gemma-data/checkpoints/gemma3-27b-pt'

GEMMA3_1B_IT = 'gs://gemma-data/checkpoints/gemma3-1b-it'

GEMMA3_4B_IT = 'gs://gemma-data/checkpoints/gemma3-4b-it'

GEMMA3_12B_IT = 'gs://gemma-data/checkpoints/gemma3-12b-it'

GEMMA3_27B_IT = 'gs://gemma-data/checkpoints/gemma3-27b-it'