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