1

私の質問は 2 つの部分に分けることができます。

  • さまざまなライブラリで使用できる機械学習モデル ファイルの形式はありますか? たとえば、モデルを pytorch で保存してから、tensorflow を使用してロードしましたか?

  • そうでない場合、pytorch 機械学習モデルを keras で直接使用できるようにフォーマットを転送するのに役立つライブラリはありますか?

この質問をする理由は、最近、テンソルフローで以前にトレーニングしたモデルの一部を pytorch に調整する必要があるためです。

この質問の更新: Facebook と Microsoft は、ONNX と呼ばれるモデル標準を立ち上げる予定です。これは、Pytorch と Caffe2 の間など、異なるフレームワーク間でモデルを転送するために使用されます。以下にリンクします。

https://research.fb.com/facebook-and-microsoft-introduce-new-open-ecosystem-for-interchangeable-ai-frameworks/

この質問のさらなる更新: Tensorflow 自体は Protocol Buffer 形式を使用してモデル ファイルを保存します。これは、異なるモデル間の転送に使用できます。以下にリンクします。

https://www.tensorflow.org/extend/tool_developers/

4

3 に答える 3

0

そうでない場合、pytorch 機械学習モデルを keras で直接使用できるようにフォーマットを転送するのに役立つライブラリはありますか?

Pytorch2Kerasコンバーターを試すことができます。

その瞬間、Conv2d、Linear、Activations、Element-wise 操作などのベース レイヤーがサポートされます。そこで、ResNet50 をエラー 1e-6 で変換しました。

于 2017-11-23T21:45:59.223 に答える