Project-MONAI / Project-MONAI/tutorials

Pretrained model not compatible with downstream training

Offen
#1,249 1 Kommentar 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen

Dieses Issue hat noch niemand übernommen.

Vorherrschende Sprache
Jupyter Notebook
Sterne
2.5k
Forks
803
Ø Merge
6 T. 22 Std.
Gemergte PRs (30 T.)
3

Beschreibung

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!!

Beitragsleitfaden

Beitragsleitfaden öffnen

Erste Schritte

  1. Lies das ganze Issue und danach den Beitragsleitfaden des Projekts.
  2. Schreib ins Issue, dass du es übernimmst — das erspart doppelte Arbeit.
  3. Forke das Repository und arbeite in einem Branch.
  4. Öffne einen Pull Request, der die Issue-Nummer nennt.

Rechercherichtung

Beginne mit dem swin-unetr pretrain repo und dem nachgelagerten Pfad für das Segmentierungstraining und verfolge dann, wie die privaten vortrainierten Encoder-Gewichte geladen werden, wenn der KeyError für module.patch_embed.proj.weight auftritt. Als abgeschlossen gilt die Feststellung, ob die vortrainierten Gewichte für die nachgelagerte Segmentierung ohne erneutes Training geladen werden können, oder die Dokumentation der fehlenden Kompatibilitätsanforderung.

Vom Indexierungsmodell aus dem Issue-Text verfasst.

Bewertung

Tech-Stack
jupyter-notebook, pytorch
Bereich
machine-learning
Issue-Typ
Bug
Schwierigkeit
4/5
Geschätzter Aufwand
3-5 Tage
Aktivitätsstatus
Veraltet
Klarheit
Muss geklärt werden
Anfängerfreundlichkeit
25/100

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.