def add(self, *layers):
with self.init_scope():
for i, layer in enumerate(layers):
index = i + len(self.__layers__)
if isinstance(layer, chainer.Link):
setattr(self, "_nn_layer_%d" % index, layer)
if isinstance(layer, Residual):
for _index, _layer in enumerate(layer.__layers__):
if isinstance(_layer, chainer.Link):
setattr(self, "_nn_layer_{}_res_{}".format(index, _index), _layer)
self.__layers__ += layers
评论列表
文章目录