在C ++中训练模型时,如何使用tensorflow C ++ API tensorflow :: ops :: SaveV2做检查点

我想知道如何使用tensorflow C ++ API:tensorflow::ops::SaveV2进行检查点。它的签名是:

SaveV2(
  const ::tensorflow::Scope & scope,::tensorflow::Input prefix,::tensorflow::Input tensor_names,::tensorflow::Input shape_and_slices,::tensorflow::InputList tensors
)

我的前缀是c ++字符串,我想保存的不同张量(Tensor对象)包括其名称(std::vector<string> names)和shapes(TensorShape对象) 。我的问题是如何将它们转换为签名中的那些不同类型。作用域参数没有问题。我的问题在于前缀,张量名称,shapes_and_slices和张量。如果有人可以编写示例代码块以列出的类型实现上述例程,我会很高兴。

liuchang1777 回答:在C ++中训练模型时,如何使用tensorflow C ++ API tensorflow :: ops :: SaveV2做检查点

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/3132850.html

大家都在问