Skip to content

Commit e6b815f

Browse files
committed
update basic layer for printing params
1 parent fafe52b commit e6b815f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorlayer/layers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -277,12 +277,12 @@ def print_params(self, details=True):
277277
try:
278278
# print(" param {:3}: {:15} (mean: {:<18}, median: {:<18}, std: {:<18}) {}".format(i, str(p.eval().shape), p.eval().mean(), np.median(p.eval()), p.eval().std(), p.name))
279279
val = p.eval()
280-
print(" param {:3}: {:20} {:15} (mean: {:<18}, median: {:<18}, std: {:<18}) ".format(i, p.name, str(val.shape), val.mean(), np.median(val), val.std()))
280+
print(" param {:3}: {:20} {:15} {} (mean: {:<18}, median: {:<18}, std: {:<18}) ".format(i, p.name, str(val.shape), p.dtype.name, val.mean(), np.median(val), val.std()))
281281
except Exception as e:
282282
print(str(e))
283283
raise Exception("Hint: print params details after tl.layers.initialize_global_variables(sess) or use network.print_params(False).")
284284
else:
285-
print(" param {:3}: {:20} {:15}".format(i, p.name, str(p.get_shape())))
285+
print(" param {:3}: {:20} {:15} {}".format(i, p.name, str(p.get_shape()), p.dtype.name))
286286
print(" num of params: %d" % self.count_params())
287287

288288
def print_layers(self):

0 commit comments

Comments
 (0)