3

pytorch を使用して画像分類モデルをトレーニングしました。
今、私はそれを研究から生産パイプラインに移したいと考えています。
TensorFlow extended の使用を考えています。PyTorch でトレーニングしたモデルを TensorFlow 拡張パイプラインで使用できるかどうかについては、まったく疑いがありません (トレーニング済みモデルを ONNX に変換してから、Tensorflow 互換フォーマットに変換できます)。オーバーヘッドが大きくなるため、トレーニング部分を TensorFlow に書き直して再トレーニングしたくありません。それは可能ですか、または PyTorch でトレーニングされたモデルを製品化するためのより良い方法はありますか?

4

2 に答える 2