gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00
Code Issues Projects Releases Wiki Activity

Printing input size as part of network summary (#310)

Browse Source
This commit is contained in:
Gal Novik
2019-05-12 15:40:02 +03:00
committed by shadiendrawis
parent ffb55b4142
commit aa9f3cefaf
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
rl_coach/architectures/tensorflow_components/embedders/embedder.py
View File

@@ -147,11 +147,11 @@ class InputEmbedder(object):
return self.name return self.name
def __str__(self): def __str__(self):
result = [] result = ['Input size = {}'.format(self._input_size)]
if self.input_rescaling != 1.0 or self.input_offset != 0.0: 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.append('Input Normalization (scale = {}, offset = {})'.format(self.input_rescaling, self.input_offset))
result.extend([str(l) for l in self.layers_params]) result.extend([str(l) for l in self.layers_params])
if self.layers_params: if not self.layers_params:
return '\n'.join(result) result.append('No layers')
else:
return 'No layers' return '\n'.join(result)