Skip to content

Commit 44a2f7c

Browse files
committed
update basic layer for print layers
1 parent 779021d commit 44a2f7c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tensorlayer/layers.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -270,7 +270,6 @@ def __init__(
270270
if name not in ['', None, False]:
271271
set_keep['_layers_name_list'].append(name)
272272

273-
274273
def print_params(self, details=True):
275274
''' Print all info of parameters in the network'''
276275
for i, p in enumerate(self.all_params):
@@ -286,8 +285,9 @@ def print_params(self, details=True):
286285

287286
def print_layers(self):
288287
''' Print all info of layers in the network '''
289-
for i, p in enumerate(self.all_layers):
290-
print(" layer %d: %s" % (i, str(p)))
288+
for i, layer in enumerate(self.all_layers):
289+
# print(" layer %d: %s" % (i, str(layer)))
290+
print(" layer {:3}: {:20} {:15} {}".format(i, layer.name, str(layer.get_shape()), layer.dtype.name))
291291

292292
def count_params(self):
293293
''' Return the number of parameters in the network '''

0 commit comments

Comments
 (0)