How to access self.log on_init_() or before_train()? #11641
Unanswered
allanchan339
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 1 comment 4 replies
-
have you taken a look at the model summary callback? https://pytorch-lightning.readthedocs.io/en/latest/extensions/generated/pytorch_lightning.callbacks.ModelSummary.html#pytorch_lightning.callbacks.ModelSummary that's an example of something which runs before training and can include the model size, parameters per module, and whose output can be logged to another system. by default it prints out a table You can do something once before training in the |
Beta Was this translation helpful? Give feedback.
4 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.
-
By pl docs, callbacks to implement unnecessary research code are recommended. Therefore, I would like to record the total trainable parameters numbers, total size, and another record in the logging system.
However, it seems to me that the self.log can only be accessed after training is initialized.
Any better way to access the log once only before training?
Beta Was this translation helpful? Give feedback.
All reactions