PyTorch Lightning验证步骤日志未覆盖问题咨询

训练阶段的进度条默认用了控制台行覆盖（通过回车符\r实现），每次更新进度时会回到行首覆盖之前的内容，所以最终只保留100%的状态；而验证阶段的日志是每更新一次就输出新行，所有进度记录都能留存。

从日志里的v_num、global step等关键词判断，你大概率在用PyTorch Lightning，以下分场景给出调整方案：

1. 基于PyTorch Lightning的调整 #

• 直接修改Trainer参数，强制训练进度条保留所有更新记录：

  from pytorch_lightning import Trainer
  from pytorch_lightning.callbacks import RichProgressBar
  
  trainer = Trainer(
      # 让进度条每步都输出新行，避免覆盖
      progress_bar_refresh_rate=1,
      # 用Rich进度条并设置leave=True，保留所有进度行
      callbacks=[RichProgressBar(leave=True)]
  )
  

• 也可以禁用默认进度条，手动在训练循环中打印进度：

  def training_step(self, batch, batch_idx):
      total_batches = len(self.train_dataloader)
      progress = (batch_idx + 1) / total_batches * 100
      print(f"Epoch {self.current_epoch}: {progress:.0f}%|{(batch_idx + 1)}/{total_batches}")
      # 你的训练逻辑代码
  

2. 直接使用tqdm的情况 #

如果是手动用tqdm实现进度条，给训练迭代器设置leave=True并指定输出到标准输出：

from tqdm import tqdm
import sys

for epoch in range(num_epochs):
    # leave=True保留所有进度行，file=sys.stdout确保输出到控制台
    train_bar = tqdm(train_loader, leave=True, file=sys.stdout)
    for batch in train_bar:
        # 更新进度描述
        train_bar.set_description(f"Epoch {epoch}: {train_bar.n/train_bar.total*100:.0f}%")
        # 你的训练逻辑代码

调整后训练阶段的日志会和验证阶段一致，每段进度都会输出新行，所有百分比状态都会保留在控制台日志中。

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