gemma2_2b

torchtune.models.gemma2.gemma2_2b() → TransformerDecoder

用于创建 Gemma2 2B 模型的构建器，该模型使用以下地址的默认 2b 参数值进行初始化：https://github.com/google/gemma_pytorch/blob/main/gemma/config.py

返回:

Gemma2 2B 模型实例

返回类型:

TransformerDecoder