visualtorch

0.2.4
10.73k

Architecture visualization of Torch models