From 83ccd21beca24649f9d609761af0764a9f4adc3b Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 16:20:42 -0400 Subject: [PATCH] added sample input for summary --- .../examples/new_project_templates/lightning_module_template.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py index fc28c55d..e4ca0c8f 100644 --- a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py +++ b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py @@ -28,6 +28,8 @@ class LightningTemplateModel(LightningModule): self.batch_size = hparams.batch_size + self.example_input_array = torch.rand(5, 3 * 28 * 28) + # build model self.__build_model()