From 20e38a7be9c3004b84b5ddcf49399e150d6ce08c Mon Sep 17 00:00:00 2001 From: Yu-Cheng Tsai Date: Mon, 1 Apr 2024 20:23:40 -0700 Subject: [PATCH] Add file --- litgpt/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litgpt/utils.py b/litgpt/utils.py index fb6a86c107..10591a5b7c 100644 --- a/litgpt/utils.py +++ b/litgpt/utils.py @@ -451,9 +451,9 @@ def choose_logger( **kwargs: Any, ): if logger_name == "csv": - return CSVLogger(root_dir=(out_dir / "logs"), name="csv", flush_logs_every_n_steps=log_interval, **kwargs) + return CSVLogger(root_dir=(out_dir / "logs"), name=name, flush_logs_every_n_steps=log_interval, **kwargs) if logger_name == "tensorboard": - return TensorBoardLogger(root_dir=(out_dir / "logs"), name="tensorboard", **kwargs) + return TensorBoardLogger(root_dir=(out_dir / "logs"), name=name, **kwargs) if logger_name == "wandb": return WandbLogger(project=name, resume=resume, **kwargs) raise ValueError(f"`--logger_name={logger_name}` is not a valid option. Choose from 'csv', 'tensorboard', 'wandb'.")