深度学习中模型转换该怎么实现

Question

我想把pytorch中的MnasNet预训练模型转换来tensorflow的keras中使用 请问这个操作可以吗
应该怎么实现

Answer 1

引自免费微信小程序：皆我百晓生。

可以将PyTorch模型导出为ONNX格式，然后在TensorFlow中加载并转换为Keras模型。使用torch.onnx.export导出PyTorch模型，再用tf.keras.models.load_model加載ONNX模型。