From b0a1d2eb09e945eae064e4b62285434a423bb7df Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 15 Aug 2023 22:20:15 +0200
Subject: [PATCH] [engine.predictor] Do not announce tensorboard - it is
useless in this stage
---
src/ptbench/engine/predictor.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/src/ptbench/engine/predictor.py b/src/ptbench/engine/predictor.py
index 92597797..033e0829 100644
--- a/src/ptbench/engine/predictor.py
+++ b/src/ptbench/engine/predictor.py
@@ -65,11 +65,11 @@ def run(
output_folder,
log_dir,
)
- logger.info(
- f"Monitor prediction with `tensorboard serve "
- f"--logdir={output_folder}/{log_dir}/`. "
- f"Then, open a browser on the printed address."
- )
+ # logger.info(
+ # f"Monitor prediction with `tensorboard serve "
+ # f"--logdir={output_folder}/{log_dir}/`. "
+ # f"Then, open a browser on the printed address."
+ # )
accelerator, devices = device_manager.lightning_accelerator()
trainer = lightning.pytorch.Trainer(
--
GitLab