From 3da5ac995c0b325b5e37d472bedf04254197c862 Mon Sep 17 00:00:00 2001 From: dangqingqing Date: Fri, 7 Apr 2017 15:35:09 +0800 Subject: [PATCH] bug fix --- python/paddle/v2/topology.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/paddle/v2/topology.py b/python/paddle/v2/topology.py index 3e6fded839b..dbcd86b54fd 100644 --- a/python/paddle/v2/topology.py +++ b/python/paddle/v2/topology.py @@ -13,6 +13,7 @@ # limitations under the License. import collections +import copy from paddle.proto.ModelConfig_pb2 import ModelConfig @@ -66,10 +67,13 @@ class Topology(object): self.layers = layers if extra_layers is not None: extra_layers = __check__(extra_layers) - self.layers.extend(extra_layers) self.__model_config__ = v2_layer.parse_network( *layers, extra_layers=extra_layers) + + if extra_layers is not None: + self.layers.extend(extra_layers) + assert isinstance(self.__model_config__, ModelConfig) def proto(self): -- GitLab