Project-MONAI / Project-MONAI/tutorials
Pretrained model not compatible with downstream training
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
Erste Schritte
- Lies das ganze Issue und danach den Beitragsleitfaden des Projekts.
- Schreib ins Issue, dass du es übernimmst — das erspart doppelte Arbeit.
- Forke das Repository und arbeite in einem Branch.
- Ö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