When I update my PL version to 0.9, I started getting this warning:
python3.8/site-packages/pytorch_lightning/utilities/distributed.py:37: UserWarning: Could not log computational graph since the model.example_input_array attribute is not set or input_array was not given
What is the reason for this warning, and how to disable it?
Hi @teddy
Thank you for your response.
This can be useful when the example_input_array is an easy one, like in the case if image classification (the input is only 4-d tensor), but there are some cases where the input examples are very complex to create, and maybe the user is not interested in reading a big graph, especially that tensorboard become laggy when the graph has many component.
In my opinion, it’s better to make this feature optional, if the user is interested in it, he can activate it!