将pyTorch代码转换为CNTK代码

Question

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

我以前翻译过TensorFlow和CNTK代码，我发现它很容易。但我对Tensorflow相当了解...但现在我不想把精力放在学习pyTorch上。

Answer 1

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