ModelCheckpoint and WandbLogger: set a common checkpoint filename #16965
Unanswered
sh3rlock14
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 1 comment
-
This is really important... any workaround? |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hey everyone 👋🏻!
In
pytorch_lightning.callbacks.ModelCheckpoint
thefilename
parameter allows to set an highly customized filename for the checkpoint being logged, and this is super!🚀... but is it possible to pass this filename topytorch_lightning.loggers.WandbLogger
checkpoint_name
parameter?To have access to
wandb.config
I initialize WandbLogger first and ModelChekpoint then, and I would like to keep this configuration; so, since both the 'callbacks' and the 'logger' are passed to aTrainer
, wouldn't it be possible for it to update the logger's checkpoint filename with ModelCheckpoint one?Thanks in advance for your help! 🙏🏻
Beta Was this translation helpful? Give feedback.
All reactions