解读TensorBoard训练损失曲线的X轴（PyTorch-Lightning场景）

训练基本信息 #

• 数据集规模：训练集6152个样本，验证集、测试集各769个样本
• 训练代码：

  trainer = pl.Trainer(logger = logger, checkpoint_callback = checkpoint_callback, max_epochs = 10, gpus = 1, progress_bar_refresh_rate = 20)
  

• 训练轮次：10个epoch

X轴含义分析 #

如果损失曲线的X轴未显示epoch，大概率代表全局步数（global step），也就是模型累计更新参数的次数。

结合你的数据集规模，假设训练用的batch_size为N，那么单个epoch内的训练步数为 ceil(6152 / N)（向上取整，因为最后一个batch样本数可能不足N）。10个epoch的总步数就是单个epoch步数×10，曲线X轴的最大值会接近这个总步数。

举个例子：如果batch_size设为32，单个epoch步数是 6152 ÷ 32 = 192.25，向上取整为193步，10个epoch总步数就是1930，曲线X轴范围会是0~1929左右。

你可以通过核对曲线总步数和上述计算值是否匹配，来确认这个结论。

内容的提问来源于stack exchange，提问作者rana