Dear all,
I wish to plot the train and validation loss graph outside from the tensorboard.
May I know how to do it after the trainer?
Here is my code:
from pytorch_lightning.loggers import TensorBoardLogger
from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint
checkpoint_callback = ModelCheckpoint(
dirpath="speed",
filename="best-checkpoint",
save_top_k = 1,
verbose=True,
monitor="val_loss",
mode="min"
)
logger = TensorBoardLogger(“lightning_logs”, name=“test”)
early_stopping_callback = EarlyStopping(monitor=“val_loss”, patience=2)
trainer = pl.Trainer(
logger=logger,
checkpoint_callback=checkpoint_callback,
callbacks=[early_stopping_callback],
max_epochs=N_EPOCHS,
gpus=0,
progress_bar_refresh_rate=30
)
Thank you very much.