torchinfo¶
" Torchinfo provides information complementary to what is provided by print(your_model) in PyTorch, similar to Tensorflow's model.summary() API to view the visualization of the model, which is helpful while debugging your network.
homepage: https://github.com/TylerYep/torchinfo
| version | versionsuffix | toolchain |
|---|---|---|
1.5.2 |
-PyTorch-1.7.1 |
foss/2020b |
1.5.2 |
-PyTorch-1.7.1 |
fosscuda/2020b |
(quick links: (all) - 0 - a - b - c - d - e - f - g - h - i - j - k - l - m - n - o - p - q - r - s - t - u - v - w - x - y - z)