Added two new parameters to summary: verbose and csv. Verbose shows m…

sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras

MIT License

4.01k stars 412 forks source link

Added two new parameters to summary: verbose and csv. Verbose shows m… #108

Open eplatero97 opened 4 years ago

eplatero97 commented 4 years ago

…ore information from Convolutional layers and csv saves all outputs onto two csv files. The added functionalities will give the user more control over its model and a user-friendly way to extract the data