ChatGLM2-6B 训练参数解释

Num examples = 243

Num Epochs = 100 Instantaneous batch size per device = 4

Total train batch size (w. parallel, distributed & accumulation) = 16

Gradient Accumulation steps = 4

Total optimization steps = 1,500

Number of trainable parameters = 1,949,696

这些参数是深度学习模型训练过程中的一些关键设置。我来为你逐一解释:

  1. Num examples = 243: 这表示在你的训练集中有243个样本,即243个独立的数据点用于训练模型。

  2. Num Epochs = 100: 一个epoch指的是模型在训练过程中遍历整个训练集一次。因此,Num Epochs = 100意味着模型会遍历整个训练集100次。

  3. Instantaneous batch size per device = 4: 在深度学习中,我们通常不会同时处理所有的训练样本,而是将它们分成“批次”进行处理。每个批次的大小就是每次模型训练的样本数量。在这个例子中,每个设备上的即时批量大小为4,意味着每个设备一次处理4个样本。

  4. Total train batch size (w. parallel, distributed & accumulation) = 16: 这表示在并行、分布式和积累情况下,总的训练批次大小为16。这可能意味着在多个设备上同时进行训练,每个设备处理一部分批次,然后把这些批次加起来,总和为16。

  5. Gradient Accumulation steps = 4: 梯度累积是一种在内存不足的情况下训练大模型的技巧。它的工作原理是:在进行反向传播并更新模型权重之前,先计算并累积一定步数的梯度。在这个例子中,我们在每4个批次后进行一次权重更新

  6. Total optimization steps = 1,500: 优化步数是模型训练过程中权重更新的总次数。在这个例子中,模型权重将被更新1500次。

  7. Number of trainable parameters = 1,949,696: 这是模型中可以通过训练改变的参数的数量。深度学习模型的性能通常与其可训练参数的数量有关。但是,更多的参数并不总是意味着更好的性能,因为过多的参数可能导致过拟合,即模型过于复杂,不能很好地泛化到训练集之外的新数据

"add_bias_linear": false,
  "add_qkv_bias": true,
  "apply_query_key_layer_scaling": true,
  "apply_residual_connection_post_layernorm": false,
  "architectures": [
    "ChatGLMModel"
  ],
  "attention_dropout": 0.0,
  "attention_softmax_in_fp32": true,
  "auto_map": {
    "AutoConfig": "configuration_chatglm.ChatGLMConfig",
    "AutoModel": "modeling_chatglm.ChatGLMForConditionalGeneration",
    "AutoModelForSeq2SeqLM": "modeling_chatglm.ChatGLMForConditionalGeneration"
  },
  "bias_dropout_fusion": true,
  "eos_token_id": 2,
  "ffn_hidden_size": 13696,
  "fp32_residual_connection": false,
  "hidden_dropout": 0.0,
  "hidden_size": 4096,
  "kv_channels": 128,
  "layernorm_epsilon": 1e-05,
  "model_type": "chatglm",
  "multi_query_attention": true,
  "multi_query_group_num": 2,
  "num_attention_heads": 32,
  "num_layers": 28,
  "original_rope": true,
  "pad_token_id": 2,
  "padded_vocab_size": 65024,
  "post_layer_norm": true,
  "quantization_bit": 0,
  "rmsnorm": true,
  "seq_length": 32768,
  "tie_word_embeddings": false,
  "torch_dtype": "float16",
  "transformers_version": "4.30.2",
  "use_cache": true

这些参数是深度学习模型配置的详细设置,特别是对于一种称为ChatGLM的模型。以下是每个参数的含义:

  1. "add_bias_linear": false: 表示是否在线性层中添加偏置项。

  2. "add_qkv_bias": true: 表示是否在注意力机制的查询(Q)、键(K)和值(V)计算中添加偏置项。

  3. "apply_query_key_layer_scaling": true: 表示是否对注意力机制中的查询和键进行缩放处理。

  4. "apply_residual_connection_post_layernorm": false: 表示是否在层归一化后应用残差连接。

  5. "architectures": ["ChatGLMModel"]: 表示该配置用于的模型架构。

  6. "attention_dropout": 0.0: 表示在注意力计算中应用的dropout的比率。Dropout是一种防止模型过拟合的技术。

  7. "attention_softmax_in_fp32": true: 表示是否在单精度浮点格式(FP32)中执行注意力机制的Softmax计算。

  8. "auto_map": 这部分将自动配置,模型映射到ChatGLM的配置和模型。

  9. "bias_dropout_fusion": true: 表示是否融合偏置和dropout。这通常用于优化和提高训练速度。

  10. "eos_token_id": 2: 定义结束符(End of Sentence)的标识符。

  11. "ffn_hidden_size": 13696: 表示前馈神经网络(Feedforward Neural Network,FFN)的隐藏层的大小。

  12. "fp32_residual_connection": false: 表示是否在单精度浮点格式(FP32)中应用残差连接。

  13. "hidden_dropout": 0.0: 隐藏层的dropout率。

  14. "hidden_size": 4096: 隐藏层的大小。

  15. "kv_channels": 128: 键值(Key-Value)的通道数。

  16. "layernorm_epsilon": 1e-05: 层归一化的epsilon值,为了防止除数为零。

  17. "model_type": "chatglm": 模型类型。

  18. "multi_query_attention": true: 表示是否使用多查询注意力。

  19. "multi_query_group_num": 2: 在多查询注意力中的查询组数。

  20. "num_attention_heads": 32: 注意力机制的头数。

  21. "num_layers": 28: 模型的层数。

  22. "original_rope": true: 是否使用原始的ROPE模式。

  23. "pad_token_id": 2: 定义填充符的标识符。

  24. "padded_vocab_size": 65024: 表示经过填充后的词汇表大小。

  25. "post_layer_norm": true: 是否在层后应用层归一化。

  26. "quantization_bit": 0: 表示量化的位数。

  27. "rmsnorm": true: 表示是否使用RMS归一化。

  28. "seq_length": 32768: 序列长度。

  29. "tie_word_embeddings": false: 是否绑定输入和输出的词嵌入。

  30. "torch_dtype": "float16": 使用的数据类型,这里是半精度浮点数。

  31. "transformers_version": "4.30.2": 使用的Transformers库版本。

  32. "use_cache": true: 是否使用缓存以加快计算速度。

请注意,上述解释中的一些参数可能在不同的上下文和模型中有所不同,因为它们是针对特定的ChatGLM模型进行解释的。

猜你喜欢

转载自blog.csdn.net/sinat_37574187/article/details/131433629