wassname/torchsummaryX
1
0
Fork 0
mirror of https://github.com/wassname/torchsummaryX.git synced 2026-06-27 19:33:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Namhyuk Ahn
2018-07-30 09:53:27 +09:00
committed by GitHub
parent e0df04c636
commit a6ad5f983b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+3 -3
View File
@@ -9,9 +9,9 @@ from torchsummaryX import summary
summary(your_model, torch.zeros((1, 3, 224, 224)))
```
Args:
- model (Module): Model to summarize
- x (Tensor): Input tensor of the model with [N, C, H, W] shape dtype and device have to match to the model
- args, kwargs: Other arguments used in `model.forward` function
- `model` (Module): Model to summarize
- `x` (Tensor): Input tensor of the model with [N, C, H, W] shape dtype and device have to match to the model
- `args, kwargs`: Other arguments used in `model.forward` function
## Examples
CNN for MNIST