release/0.10.0
dangqingqing 8 years ago
parent edf36423fa
commit 3da5ac995c

@ -13,6 +13,7 @@
# limitations under the License. # limitations under the License.
import collections import collections
import copy
from paddle.proto.ModelConfig_pb2 import ModelConfig from paddle.proto.ModelConfig_pb2 import ModelConfig
@ -66,10 +67,13 @@ class Topology(object):
self.layers = layers self.layers = layers
if extra_layers is not None: if extra_layers is not None:
extra_layers = __check__(extra_layers) extra_layers = __check__(extra_layers)
self.layers.extend(extra_layers)
self.__model_config__ = v2_layer.parse_network( self.__model_config__ = v2_layer.parse_network(
*layers, extra_layers=extra_layers) *layers, extra_layers=extra_layers)
if extra_layers is not None:
self.layers.extend(extra_layers)
assert isinstance(self.__model_config__, ModelConfig) assert isinstance(self.__model_config__, ModelConfig)
def proto(self): def proto(self):

Loading…
Cancel
Save