Transfer learning unable to do
まだ誰も着手していません。
評価
- 難易度
- 4/5
- 見積もり時間
- 3〜5日
- 初心者へのやさしさ
- 28/100
- issue の種類
- バグ
- 明瞭さ
- 説明が足りない
- 活発さ
- 停滞
- 技術スタック
- python
調査の方向性
まず、deeplab.py の get_deeplab_resnest101_ade、ade20k.py のクラス数、train_dist.py でのモデルの読み込みと head の置換を追跡します。提供されている train_dist.py コマンドを実行し、出力クラス数がどこで決定されているかを調べます。学習済みモデルが150ではなく8クラスを生成すれば完了です。
索引モデルが issue の本文から書いたものです。
説明
Hi @zhanghang1989
I am unable to do transfer learning on the model
i have downloaded get_deeplab_resnest101_ade and
when i changed the no of classes in ade20k.py (no of classes 8)
pre trained model is not loading ( getting error )
So I have changed the code for Transfer learning
Code changes:
- deeplab.py
in get_deeplab_resnest101_ade function changed from
from
#model = DeepLabV3(datasets[dataset.lower()].NUM_CLASS, backbone=backbone, root=root, **kwargs)
to
no_of_classes = 150
model = DeepLabV3(no_of_classes, backbone=backbone, root=root, **kwargs)
so i can load pretrained model with 150 classes
then
2) In train_dist.py file
Model loading
model_ft = get_deeplab_resnest101_ade(pretrained=True)
for param in model.parameters():
param.requires_grad = False
model_ft.head.block = Sequential(
(Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)),
(BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)),
(ReLU(inplace=True)),
(Dropout(p=0.1, inplace=False)),
(Conv2d(256, 8, kernel_size=(1, 1), stride=(1, 1))))
for param in model_ft.head.parameters():
param.requires_grad = True
Training
python train_dist.py --dataset ade20k --model deeplab --aux --backbone resnest101 --ft --epochs 100
after successful training
i am getting 150 classes output not 8 classes(i have given 8 classes in last layer)
i need 8 classes output
can you help me with this
- 主要言語
- Python
- スター
- 2k
- フォーク
- 448
- PR マージ指標
- 30日以内にマージされた PR はありません
環境構築
- Dockerfile または Docker Compose ファイルあり
- プルリクエストのテンプレートなし
- コントリビューションガイドなし
はじめの一歩
- issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
- 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
- リポジトリをフォークし、ブランチを切って変更します。
- issue 番号を参照したプルリクエストを送ります。
zhanghang1989/PyTorch-Encoding のほかの issue
-
難易度 3/5 1〜2日 初心者へのやさしさ 30/100
zhanghang1989/PyTorch-Encoding#432 · コメント 2 件 ·
-
難易度 3/5 1〜2日 初心者へのやさしさ 25/100
zhanghang1989/PyTorch-Encoding#430 · コメント 3 件 · リアクション 2 件 ·
-
,オープン
難易度 5/5 1週間以上 初心者へのやさしさ 1/100
-
難易度 4/5 3〜5日 初心者へのやさしさ 25/100
-
難易度 4/5 3〜5日 初心者へのやさしさ 25/100
zhanghang1989/PyTorch-Encoding#426 · コメント 1 件 ·
zhanghang1989/PyTorch-Encoding の issue をすべて見る
似ている issue
-
難易度 2/5 1〜3時間 初心者へのやさしさ 64/100
aws-samples/sample-per-model-bedrock#295 ·
メンテナーはふだん 1 日以内に返信
-
難易度 1/5 1時間未満 初心者へのやさしさ 76/100
pyrevitlabs/pyRevit#3792 ·
メンテナーはふだん 1 日以内に返信
-
難易度 2/5 1〜3時間 初心者へのやさしさ 69/100
zephyrproject-rtos/reqmgmt#177 ·
-
難易度 2/5 1〜3時間 初心者へのやさしさ 78/100
polarsource/polar#15580 ·
メンテナーはふだん 1 日以内に返信
-
難易度 1/5 1時間未満 初心者へのやさしさ 85/100
521xueweihan/HelloGitHub#4011 ·