解释 TensorBoard 中的损失曲线

Question

我使用 pytorch-lightning 训练器来训练我的模型。训练集有6152，测试集和验证集有769样本。训练完成了 10 个时期.

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

火车损失曲线

Y 轴显示训练损失。 X 轴应显示纪元 (0, 1, 2, ..., 9)，此处不可用。请帮助理解这条曲线的 X 轴。

Answer 1

X 轴是步数。在每个时代，你要走 N 步。 （如果您在 batch_size = 4 的 1 个 GPU 上运行训练）N = training_samples/4。所以最大步数就是 n_epochs * N.