在同一行的pytorch点燃ProgressBar上写入验证度量

Question

是否有一种方法可以像在keras中一样使用pytorch ignite在进度条的末尾编写验证度量？

现在，我正在使用ProgressBar.log_message(下面的示例(来打印相关的度量，但这样做会在每个epoch的末尾打印一行新行。我希望消息打印在栏的末尾，不带换行符。

@trainer.on(Events.EPOCH_COMPLETED)
def run_evaluator(trainer):
evaluator.run(val_loader)
val_metrics = evaluator.state.metrics
# ProgBar
pbar.log_message(f"Val_loss: {val_metrics['Val_loss']:.2f}")
pbar.log_message(f"RMSE: {val_metrics['RMSE']:.2f}")

为了进一步澄清，以下是我现在得到的输出：

Epoch [1/3]: [3/3] 100%|██████████| [00:00/00:00], Train_loss=1.59e+4
Val_loss: 26333.96
RMSE: 72572.66
Epoch [2/3]: [3/3] 100%|██████████| [00:00/00:00], Train_loss=2.35e+4
Val_loss: 25508.46
RMSE: 71426.14
Epoch [3/3]: [3/3] 100%|██████████| [00:00/00:00], Train_loss=2.26e+4
Val_loss: 24721.82
RMSE: 70316.17

这就是我想要的：

Epoch [1/3]: [3/3] 100%|██████████| [00:00/00:00], Train_loss=1.59e+4  Val_loss: 26333.96  RMSE: 72572.66
Epoch [2/3]: [3/3] 100%|██████████| [00:00/00:00], Train_loss=2.35e+4  Val_loss: 25508.46  RMSE: 71426.14
Epoch [3/3]: [3/3] 100%|██████████| [00:00/00:00], Train_loss=2.26e+4  Val_loss: 24721.82  RMSE: 70316.17

Answer 1

似乎与tqdm有关，而不是与点火有关。请在github或pytorch讨论论坛上询问其他问题，我们的开发团队不在SO上。谢谢！