无法在张量流中还原模型检查点

时间:2019-05-10 18:24:51

标签: python tensorflow

我已经从以下代码运行样式转换模型:https://github.com/eridgd/WCT-TF。当我在终端中运行它时,这工作正常。但是,当我在jupyter笔记本中运行相同的代码时,它会显示“在检查点中未找到键编码器_decoder_relu5_1_1 / decoder_relu5_1 / decoder_model_relu5_1 / relu5_1_0 / relu5_1_0 /偏差”

但是我在jupyter笔记本中使用与终端相同的检查点。

感谢您的帮助!

我在这里阅读了该帖子:https://github.com/tensorflow/models/issues/466。但是不确定是否是相同的问题,因为我可以在终端中毫无问题地执行代码。

def main():
    start = time.time()
    cp = ["models/relu5_1", "models/relu4_1",
          "models/relu3_1", "models/relu2_1",
          "models/relu1_1"]
    relu_ = ["relu5_1", "relu4_1", "relu3_1", "relu2_1", "relu1_1"]
    # Load the WCT model
    wct_model = wct.WCT(checkpoints=cp, 
                    relu_targets=relu_,
                    vgg_path='C:\\Users\\zyu\\Desktop\\Style Transfer\\models\\vgg_normalised.t7',
                    ss_patch_size=3, ss_stride=1
                   )

此wct.WCT函数就是问题所在。

0 个答案:

没有答案