From a6ad5f983bd4d808d360d55fc7e34105cdce215b Mon Sep 17 00:00:00 2001 From: Namhyuk Ahn Date: Mon, 30 Jul 2018 09:53:27 +0900 Subject: [PATCH] Update README.md --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 003a792..e9c8ef1 100644 --- a/README.md +++ b/README.md @@ -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