Hi @aouie,
It looks like it’s related to this post. The solution seems to be to do a forward pass through the model before printing the summary.
Let me know if this helps!