I still have this problem. The tool finishes, but nothing gets printed.
I am using torch 1.7.1
Is it possible that all exceptions are caught?
Btw my network is quite deep. Torchviz recently throwed an exception because recursion depth was reached.