Dear users, we have installed a new software: torchinfo 1.5.2-PyTorch-1.11.0-CUDA-11.3.1:
---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
torchinfo: torchinfo/1.5.2-PyTorch-1.11.0-CUDA-11.3.1
---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
Description:
" 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.
You will need to load all module(s) on any one of the lines below before the "torchinfo/1.5.2-PyTorch-1.11.0-CUDA-11.3.1" module is available to load.
GCC/10.3.0 OpenMPI/4.1.1
Help:
Description
===========
"
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.
More information
================
- Homepage: https://github.com/TylerYep/torchinfo
Best,
HPC team