How to define example_input_array for computational graph #10405
Unanswered
minwang-ai
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 2 comments 6 replies
-
you can simply assign it to a variable: class LitModel(LightningModule):
def __init__(self, ...):
...
self.example_input_array = ... |
Beta Was this translation helpful? Give feedback.
6 replies
-
Not for me. |
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
-
Could you please give me an example for defining self.example_input_array according to the document?
When and where should I log computational graph, train/Val/test step?
log_graph (bool) – Adds the computational graph to tensorboard. This requires that the user has defined the self.example_input_array attribute in their model.
Beta Was this translation helpful? Give feedback.
All reactions