2020年12月16日
深度学习模型转换，以pytorch转tensorflow为例
摘要： 这里以onnx为中介进行转换。主要用到 STEP1. 将pytorch 模型转换成onnx模型 注意这里关键是要构造一个模型的输入输入，这里假设模型接受两个输入。 pmodel = PytorchModel() dummy_input = (np.zeros((1, 30), dtype=np.fl 阅读全文
posted @ 2020-12-16 19:50 mrbean 阅读(3942) 评论(0) 推荐(0) 编辑