Skip to content

How to set experiment name such that it can be some unique name instead of version_0, ... etc. #9185

Discussion options

You must be logged in to vote

You can pass Trainer a custom logger with the version specified.

from pytorch_lightning.loggers import TensorBoardLogger

logger = TensorBoardLogger("default_root_dir", version="your_version", name="my_model")
trainer = Trainer(logger=logger)

Here is the api of TensorBoardLogger

Replies: 1 comment 2 replies

Comment options

You must be logged in to vote
2 replies
@dazzle-me
Comment options

@tshu-w
Comment options

Answer selected by carmocca
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment