本文整理汇总了Python中layer.Layer.load_weight方法的典型用法代码示例。如果您正苦于以下问题:Python Layer.load_weight方法的具体用法?Python Layer.load_weight怎么用?Python Layer.load_weight使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类layer.Layer
的用法示例。
在下文中一共展示了Layer.load_weight方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: build_net_from_copy
# 需要导入模块: from layer import Layer [as 别名]
# 或者: from layer.Layer import load_weight [as 别名]
def build_net_from_copy(self, copy):
"""
Rebuild the net from a copy made by make_copy.
"""
nnstore = copy
self.num_layers = len(nnstore.layer)
self.layer = []
for i in range(self.num_layers):
in_dim, out_dim = nnstore.layer[i].W.shape
new_layer = Layer(in_dim, out_dim, nnstore.layer[i].act_type)
new_layer.load_weight(nnstore.layer[i].W, nnstore.layer[i].b)
self.layer.append(new_layer)
in_dim, out_dim = nnstore.output.W.shape
new_layer = OutputLayer(in_dim, out_dim, nnstore.output.act_type)
new_layer.load_weight(nnstore.output.W, nnstore.output.b)
self.output = new_layer
if self.num_layers > 0:
self.input_dim = self.layer[0].W.shape[0]
else:
self.input_dim = self.output.W.shape[0]