Pardon me, I cannot replicate the errors on my project.
I changed the model structure, recreate the virtualenv envrionment, then I run the jupyter-notebook a few seconds ago and got all thing worked right, the model could be visualized, displaed on jupyter-notebook or saved as .png format on my disk.
I’m not sure whether there was wrong with the installation of pytorch or flaws existed in the model at that time.
But I’m sure the torchsummary could be executed to display the network channels and sizes any time.
As the problem disappeared, shall I create an issue on GitHub?