Visualize the whole network for SkeletonGait++

[nyanmn]

System information (version)

• Pytorch => 2.3.0
• Operating System / Platform => Ubuntu 22.04
• Cuda => 12.0

Detailed description

I am using SkeletonGait++ and like to visualize the whole model. Try to print model summary in main.py. But I have difficulty to find the input shape. Used Netron and load trained model, but see only in modules. May I know how to visualize the net?

Issue submission checklist

• [ yes] I checked the problem with documentation, FAQ, issues, and have not found solution.