Comments (3)
from research-contributions.
Hi @davidkvcs , Thanks for reporting this issue. The pre-trained model checkpoints should contain three keys {'epochs': , 'best_acc', 'state_dict': }
Key "state_dict" should be the model weights, could you direct me to the model checkpoints you've tried, I will double check whether the saved model has consistent key-value items.
Thank you.
from research-contributions.
Hi @tangy5
I trained the model my self, using this repo.
Do you have some way I can share this .pth file with you?
Since the next line, model.load_state_dict(model_dict)
works fine and the prediction look pretty good I assume, the weights are loaded correctly with the changes I made?
It seems my trained model does not have the three keys {'epochs': , 'best_acc', 'state_dict': }, but holds the state_dict only. When I run model_dict = torch.load(pretrained_pth), model_dict is a OrderedDict of length 159, holding what could look like the trained model weights. Examples of field names:
...
swinViT.patch_embed.proj.weight
swinViT.patch_embed.proj.bias
swinViT.layers1.0.blocka.0.norm1.weight
and so on...
Does that make sense?
from research-contributions.
Related Issues (20)
- Swin-UNETR pretraining model weights do not load into the model
- Validation
- One problem came when I was using SwinUNETR to train my own data HOT 5
- Invalid organ segmentation nii file produced from prostate_mri_lesion_seg_app
- SwinUNETR-V2
- About Test HOT 4
- about evaluation HOT 1
- About SwimUNETR Model Structure
- Unnecessary conv3 Layer Weights in DAE with MONAI v0.8.0 Leading to Compatibility Issues with v0.9.0+
- FileNotFoundError in swinunetr
- Swin-UNETR pretraining validation criterion
- AttributeError: 'NoneType' object has no attribute 'items' in Auto3dSeg
- Not reproducible results
- AttributeError: 'NoneType' object has no attribute 'items' in swinunetr algo
- Brats2020 HOT 3
- Evaluation
- difference brats2021 train test
- Unexpected key(s) in state_ dict: "epoch", "best_acc", "state_dict". HOT 3
- N foreground 0, N background 52003401,unable to generate class balanced samples. NaN or Inf found in input tensor.
- Understanding the contrastive loss implementation HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from research-contributions.