Summary and visualization of the PyTorch models
Labels: Contribution Welcome
❓ Questions & Help
<!-- A clear and concise description of the question. -->Hi, I am using GCNModel and GATModel. May I ask, is there a way to summarize and visualize the structures of these models such as plot_model in Keras, please? Thanks a lot!