相关疑难解决方法(0)

如何使用不同名称但形状相同的Tensorflow恢复重量?

我在Tensorflow中有多种架构。其中一些共享某些零件的设计。

我想训练一个网络,并在另一个网络中使用经过训练的相似层的权重。

此时,我可以保存所需的权重,并将其重新加载到具有与变量完全相同的命名约定的体系结构中。

但是,当权重在两个网络中具有不同的名称时,将无法还原。对于第一个网络,我有以下命名约定:

  • selector_network / c2w / var1

在第二个网络中,我有这个:

  • joint_network / c2w / var1

除此之外,变量在形状方面相似。重新加载时是否可以更改名称,或告诉Tensorflow在哪里适合这些变量?

编辑:我从@batzner找到了这个脚本,该脚本允许重命名Tensorflow检查点的变量:tensorflow_rename_variables

它不起作用。我收到以下错误:

ValueError: Couldn't find 'checkpoint' file or checkpoints in given directory ./joint_pos_tagger_lemmatizer/fi/
Run Code Online (Sandbox Code Playgroud)

machine-learning tensorflow

6
推荐指数
1
解决办法
2519
查看次数

标签 统计

machine-learning ×1

tensorflow ×1