visualtorch

0.2.4
15.71k

Architecture visualization of Torch models