如何解决从基于onnx的模型的SavedModel重新配置Tensorflow模型
我有一个来自Pytorch的预训练的Dense-121模型,并将其导出为onnx格式。之后,我加载onnx文件,并使用onnx_tf.backend中的prepare函数。我终于用了 tf_rep.export_graph('model_var.pb'),以便我可以将onnx文件转换为基于SavedModel的tensorflow模型。 到现在为止还挺好。 我想要做的是加载此SavedModel类型文件(.pb),以获得模型配置,以便可以在Tensorflow中重新配置它,而不再是Pytorch。但是我找不到任何办法。 例如,我不能使用model.summary()。 有办法做到这一点,还是我只是在浪费时间? 如果不是,那可能是因为我使用这些工具的方式是错误的。如何在Tensorflow中重新配置用Pytorch编写的模型? 谢谢!
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。