CATEGORY

Deep Learning

PyTorchでモデルの枝刈りを行い軽量なモデルを作成する

参考情報 下記の記事を参考にしました。 https://pytorch.org/tutorials/intermediate/pruning_tutorial.html Google Colabで動作確認をしました。設定方法は下記記事に記述しています。 モデルの作成 下記のコードでLeNetのモデルを作成します。 モデルの重みとバイアスを確認します。 下記のようにパラメータを確認できます。 モデル […]

PyTorchのTorchVisionで用意されている学習済みモデルを調べてみた(識別、物体検出、セグメンテーション、ポーズ推定)

モデルの学習をせずにかんたんにモデルを試したい 深層学習ではモデルの学習をしないとモデルの動作を確認できません。ですがPyTorchでは事前学習済みモデルが用意されており、そのバリエーションが多数あります。 下記のタスクのモデルが用意されています。 識別 https://pytorch.org/docs/stable/torchvision/models.html#classification セ […]

Close Bitnami banner
Bitnami