将pyTorch代码转换为CNTK代码

Question

我需要从pyTorch重写一些代码到CNTK。 我非常了解CNTK和深度学习基础知识。 是否容易将pyTorch和CNTK联系起来？ 我需要注意一些特殊的事情吗？

之前我确实翻译了TensorFlow和CNTK代码，我觉得很容易。 但我知道Tensorflow相当不错......但现在我不想付出努力学习pyTorch。

Answer 1

您可以尝试的一件事是从PyTorch将模型导出为ONNX格式。然后你可以在CNTK中加载它。请参阅here进行保存，here进行加载。请注意，这仍然是一项非常新的工作，不同的工具包对ONNX有不同程度的支持。