From aa9f3cefaf829455b8b19d22a3156b026c466a81 Mon Sep 17 00:00:00 2001 From: Gal Novik Date: Sun, 12 May 2019 15:40:02 +0300 Subject: [PATCH] Printing input size as part of network summary (#310) --- .../tensorflow_components/embedders/embedder.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/rl_coach/architectures/tensorflow_components/embedders/embedder.py b/rl_coach/architectures/tensorflow_components/embedders/embedder.py index 5bcdef2..13544c9 100644 --- a/rl_coach/architectures/tensorflow_components/embedders/embedder.py +++ b/rl_coach/architectures/tensorflow_components/embedders/embedder.py @@ -147,11 +147,11 @@ class InputEmbedder(object): return self.name def __str__(self): - result = [] + result = ['Input size = {}'.format(self._input_size)] if self.input_rescaling != 1.0 or self.input_offset != 0.0: result.append('Input Normalization (scale = {}, offset = {})'.format(self.input_rescaling, self.input_offset)) result.extend([str(l) for l in self.layers_params]) - if self.layers_params: - return '\n'.join(result) - else: - return 'No layers' + if not self.layers_params: + result.append('No layers') + + return '\n'.join(result)