def _add_params(self, params):
if isinstance(params, list):
self.all_params.extend(list(params))
else:
self.all_params.append(params)
self.all_params = list_remove_repeat(self.all_params)
def _add_layers(self, layers):
if isinstance(layers, list):
try: # list of class Layer
new_layers = [layer.outputs for layer in layers]
self.all_layers.extend(list(new_layers))
except AttributeError: # list of tf.Tensor
self.all_layers.extend(list(layers))
else:
self.all_layers.append(layers)
self.all_layers = list_remove_repeat(self.all_layers)