如何将模型文件中的存储格式NCHW改为NHWC
收藏
我现在有训练好的模型文件和参数
想重新保存为NHWC格式 应该怎么做呢
0
收藏
请登录后评论
我的理解是,每个框架有自己的顺序。
不知道你具体要做什么,也许可以考虑通过转ONNX,再转其他格式方法迂回一下。
我想将模型里的tensor变为NHWC计算 而不是NCHW计算
ONNX之前没使用过 我再看看。谢谢啦
你最后转换过来了吗?我也要转,但不知道怎么操作