質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.50%
深層学習

深層学習は、多数のレイヤのニューラルネットワークによる機械学習手法。人工知能研究の一つでディープラーニングとも呼ばれています。コンピューター自体がデータの潜在的な特徴を汲み取り、効率的で的確な判断を実現することができます。

PyTorch

PyTorchは、オープンソースのPython向けの機械学習ライブラリ。Facebookの人工知能研究グループが開発を主導しています。強力なGPUサポートを備えたテンソル計算、テープベースの自動微分による柔軟なニューラルネットワークの記述が可能です。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

Q&A

解決済

2回答

7601閲覧

自作データセットの学習済みモデルを利用したファインチューイング

shu214

総合スコア18

深層学習

深層学習は、多数のレイヤのニューラルネットワークによる機械学習手法。人工知能研究の一つでディープラーニングとも呼ばれています。コンピューター自体がデータの潜在的な特徴を汲み取り、効率的で的確な判断を実現することができます。

PyTorch

PyTorchは、オープンソースのPython向けの機械学習ライブラリ。Facebookの人工知能研究グループが開発を主導しています。強力なGPUサポートを備えたテンソル計算、テープベースの自動微分による柔軟なニューラルネットワークの記述が可能です。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

0グッド

0クリップ

投稿2021/01/02 17:12

前提・実現したいこと

自作データセットで学習したモデルを使用してさらに別途用意してある自作データセットでファインチューニングを行い、その効果を確かめたいと思っています。

近い例としてpytorchにすでに用意されているVGGやResNetなどのネットワークおよびImageNetで学習された学習済みモデルを使用するものは多数公開されていますが、自作の学習済みモデルを使う方法がわかりません。
どなたかご教授いただけないでしょうか。
参考になる資料などご存じの方は教えていただけると幸いです。

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答2

0

自己解決に至りました。
参考URLを教えてくださった方ありがとうございました!

ネットワークを定義したあと、そのまま保存済みの学習モデルを読み込むことでできるようです。

python

1model_path = 'PATH' 2torch.save(modl.state_dict(), model_path)

python

1model = Net() 2model.load_state_dict(torch.load(model_path))

ただし、転移学習なのでどの層を再学習するのかを決める必要がある場合もあります。
そのようなときは再学習しない層を’凍結’することによってそれを実現します。

python

1for pram in model.conv1.parameters(): 2 param.requires_grad = False

例として、CNNの最初の畳み込み層を凍結しする場合、上記のように畳み込み層のパラメータを取得してrequires.gradをFalseにします。(上記コードを学習済みモデルをロードした後に挿入)

パラメータチューニングの方法はまた別にあると思われます。

確証はあまりありませんので、間違いがあればご指摘お願いします、、

投稿2021/01/03 14:54

shu214

総合スコア18

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

0

ベストアンサー

こちらの記事が参考になりそうです
Pytorchでモデルの保存と読み込み - Pythonいぬ

投稿2021/01/03 11:55

A_kirisaki

総合スコア2853

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

shu214

2021/01/03 14:37

ありがとうございます! 拝見させていただきます!
guest

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.50%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問