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