From 8db8cd25394922c5fd8925a66881de71d72d389a Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 16:28:55 -0400 Subject: [PATCH] added sample input for summary --- pytorch_lightning/root_module/memory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/root_module/memory.py b/pytorch_lightning/root_module/memory.py index ed8854ba30..a2feb689bd 100644 --- a/pytorch_lightning/root_module/memory.py +++ b/pytorch_lightning/root_module/memory.py @@ -41,7 +41,7 @@ class ModelSummary(object): if self.model.trainer.use_amp: input_ = input_.half() - with torch.no_grad: + with torch.no_grad(): for i in range(1, len(mods)): m = mods[i]