def print_layers(self):
""" Print all info of layers in the network """
for i, p in enumerate(self.all_layers):
print(" layer %d: %s" % (i, str(p)))
def count_params(self):
""" Return the number of parameters in the network """
After Change
""" Print all info of layers in the network """
for i, layer in enumerate(self.all_layers):
// print(" layer %d: %s" % (i, str(layer)))
print(" layer {:3}: {:20} {:15} {}".format(i, layer.name, str(layer.get_shape()), layer.dtype.name))
def count_params(self):
""" Return the number of parameters in the network """