Project-MONAI / Project-MONAI/tutorials
Pretrained model not compatible with downstream training
まだ誰も着手していません。
- 主要言語
- Jupyter Notebook
- スター
- 2.5k
- フォーク
- 803
- 平均マージ
- 6日 22時間
- マージ済み PR(30日)
- 3
説明
I am having some issues using a private pretrained encoder weights. When following the swin-unetr pretrain repo to train a pretrained encoder weights, the model can't be loaded for downstream segmentation training. Here is the error:
KeyError: 'module.patch_embed.proj.weight'
Is there anything that might be missing from the pretrain repo? Would it be possible to fix this issue without retraining? Thank you!!
コントリビューションガイド
はじめの一歩
- issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
- 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
- リポジトリをフォークし、ブランチを切って変更します。
- issue 番号を参照したプルリクエストを送ります。
調査の方向性
swin-unetr pretrain repo と downstream のセグメンテーション学習パスから始め、module.patch_embed.proj.weight に対する KeyError が発生したときに、非公開の事前学習済み encoder の重みがどのように読み込まれるかを追跡します。完了条件は、再学習なしで downstream のセグメンテーション用に事前学習済みの重みを読み込めるかどうかを判断すること、または不足している互換性要件を文書化することです。
索引モデルが issue の本文から書いたものです。
評価
- 技術スタック
- jupyter-notebook, pytorch
- 領域
- machine-learning
- issue の種類
- バグ
- 難易度
- 4/5
- 見積もり時間
- 3〜5日
- 活発さ
- 停滞
- 明瞭さ
- 説明が足りない
- 初心者へのやさしさ
- 25/100