【问题标题】:How to load a pre-trained tensorflow model and then inference by torch?如何加载预训练的张量流模型,然后通过火炬进行推理?
【发布时间】:2021-03-01 10:56:06
【问题描述】:

tensorflow模型和torch模型中的所有参数并行化有什么技巧吗?

【问题讨论】:

  • 请详细说明,举个例子

标签: tensorflow deep-learning pytorch torch


【解决方案1】:

您需要在 pytorch 中重新实现模型,然后加载权重。好像没有自动转换的方法。

另见数据科学堆栈交换:

【讨论】:

  • 但似乎(例如 RNN 在 torch 和 tf 中的实现是不同的,我无法将 tf 中的所有权重映射到 torch 层)
猜你喜欢
  • 1970-01-01
  • 1970-01-01
  • 2017-09-22
  • 1970-01-01
  • 1970-01-01
  • 2021-12-04
  • 1970-01-01
  • 1970-01-01
  • 2021-06-18
相关资源
最近更新 更多