我分两步训练模型:
第1步,训练Model1,可训练图层为conv1
:
input1 conv1 output1
第2步,在模型1中训练模型2,权重为conv1
(模型1已经过训练)。此处,可训练图层仅为fc1
(假设我们的权重为fc2
):
input2 fc1 conv1 output2 loss
fc2 conv1 output3
Model2中的两个conv1
位于不同的范围内并且具有不同的输入(fc1
和fc1
),但在保存conv1
时我只能定义一个名称型号1。
如何在Model2的图表中多次将conv1
的权重(如上所示使用两次)恢复?