Tensorflow在另一个模型中导入pretrain模型密集层权重

时间:2018-03-08 13:43:43

标签: python tensorflow graph tensor

我目前正在使用一个受张量流训练的模型,我希望在给定的密集层中提取权重并将其添加到我想要训练的新模型中。我的第一个模型的输出为2级但我希望从密集层获得第一个权重张量,并在训练之前将这些值添加到新模型中。这可能吗?

这是我想要做的图片

Model 1 and Model 2

0 个答案:

没有答案