Project-MONAI / Project-MONAI/tutorials

Pretrained model not compatible with downstream training

オープン
#1,249 コメント 1 件 リアクション 0 件 担当者 0 名 GitHub で見る

まだ誰も着手していません。

主要言語
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!!

コントリビューションガイド

コントリビューションガイドを開く

はじめの一歩

  1. issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
  2. 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
  3. リポジトリをフォークし、ブランチを切って変更します。
  4. 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

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。