1 Repositories
Python torchinfo Libraries
View model summaries in PyTorch!
torchinfo (formerly torch-summary) Torchinfo provides information complementary to what is provided by print(your_model) in PyTorch, similar to Tensor
1.5k Jan 5, 2023