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)