nikkie-ftnextの日記
id:nikkie-ftnext
週末ログ | GKEでKFServingを使ってPyTorchのモデルのサーブを試しハマりました(後編:デプロイしたら暗中模索)
はじめに 頑張れば、何かがあるって、信じてる。nikkieです。 週末ログの後編です。 PyTorchでモデルを訓練した後、KFServingでサーブする部分です。 目次 はじめに 目次 1. KFServingが使える環境を用意する 2. PyTorchでテキスト分類するモデルを用意 Python環境 モデルの学習スクリプト サーブするモデルに渡すテンソル…